From 3b9afd4cd2efcfeb9699f1d9658c8509e08e58fb Mon Sep 17 00:00:00 2001 From: AlexeyAB Date: Mon, 16 Jan 2017 00:44:41 +0300 Subject: [PATCH] Fixed behavior if missing library cudnn.lib --- build/darknet/darknet.vcxproj | 2 +- src/convolutional_kernels.cu | 2 ++ src/convolutional_layer.c | 2 ++ 3 files changed, 5 insertions(+), 1 deletion(-) diff --git a/build/darknet/darknet.vcxproj b/build/darknet/darknet.vcxproj index 9f3e53cf..8c0ce25e 100644 --- a/build/darknet/darknet.vcxproj +++ b/build/darknet/darknet.vcxproj @@ -136,7 +136,7 @@ true true C:\opencv_2.4.9\opencv\build\x64\vc12\lib;$(CUDA_PATH)lib\$(PlatformName);$(cudnn)\lib\x64;%(AdditionalLibraryDirectories) - ..\..\3rdparty\lib\x64\pthreadVC2.lib;cublas.lib;curand.lib;cudart.lib;cudnn.lib;%(AdditionalDependencies) + ..\..\3rdparty\lib\x64\pthreadVC2.lib;cublas.lib;curand.lib;cudart.lib;%(AdditionalDependencies) $(OutDir)\..\$(TargetName)$(TargetExt) diff --git a/src/convolutional_kernels.cu b/src/convolutional_kernels.cu index ae9df8f0..4191efec 100644 --- a/src/convolutional_kernels.cu +++ b/src/convolutional_kernels.cu @@ -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" diff --git a/src/convolutional_layer.c b/src/convolutional_layer.c index 37211ab7..798d9c16 100644 --- a/src/convolutional_layer.c +++ b/src/convolutional_layer.c @@ -8,6 +8,8 @@ #include #include +#pragma comment(lib, "cudnn.lib") + #ifdef AI2 #include "xnor_layer.h" #endif