mirror of https://github.com/AlexeyAB/darknet.git
Fixed behavior if missing library cudnn.lib
This commit is contained in:
parent
0c30a743bd
commit
3b9afd4cd2
|
@ -136,7 +136,7 @@
|
|||
<EnableCOMDATFolding>true</EnableCOMDATFolding>
|
||||
<OptimizeReferences>true</OptimizeReferences>
|
||||
<AdditionalLibraryDirectories>C:\opencv_2.4.9\opencv\build\x64\vc12\lib;$(CUDA_PATH)lib\$(PlatformName);$(cudnn)\lib\x64;%(AdditionalLibraryDirectories)</AdditionalLibraryDirectories>
|
||||
<AdditionalDependencies>..\..\3rdparty\lib\x64\pthreadVC2.lib;cublas.lib;curand.lib;cudart.lib;cudnn.lib;%(AdditionalDependencies)</AdditionalDependencies>
|
||||
<AdditionalDependencies>..\..\3rdparty\lib\x64\pthreadVC2.lib;cublas.lib;curand.lib;cudart.lib;%(AdditionalDependencies)</AdditionalDependencies>
|
||||
<OutputFile>$(OutDir)\..\$(TargetName)$(TargetExt)</OutputFile>
|
||||
</Link>
|
||||
<CudaCompile>
|
||||
|
|
|
@ -2,6 +2,8 @@
|
|||
#include "curand.h"
|
||||
#include "cublas_v2.h"
|
||||
|
||||
#pragma comment(lib, "cudnn.lib")
|
||||
|
||||
extern "C" {
|
||||
#include "convolutional_layer.h"
|
||||
#include "batchnorm_layer.h"
|
||||
|
|
|
@ -8,6 +8,8 @@
|
|||
#include <stdio.h>
|
||||
#include <time.h>
|
||||
|
||||
#pragma comment(lib, "cudnn.lib")
|
||||
|
||||
#ifdef AI2
|
||||
#include "xnor_layer.h"
|
||||
#endif
|
||||
|
|
Loading…
Reference in New Issue