Fixed behavior if missing library cudnn.lib

This commit is contained in:
AlexeyAB 2017-01-16 00:44:41 +03:00
parent 0c30a743bd
commit 3b9afd4cd2
3 changed files with 5 additions and 1 deletions

View File

@ -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>

View File

@ -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"

View File

@ -8,6 +8,8 @@
#include <stdio.h>
#include <time.h>
#pragma comment(lib, "cudnn.lib")
#ifdef AI2
#include "xnor_layer.h"
#endif