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

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