mirror of
https://github.com/pjreddie/darknet.git
synced 2023-08-10 21:13:14 +03:00
Fixed behavior if missing library cudnn.lib
This commit is contained in:
@ -2,7 +2,9 @@
|
||||
#include "curand.h"
|
||||
#include "cublas_v2.h"
|
||||
|
||||
#ifdef CUDNN
|
||||
#pragma comment(lib, "cudnn.lib")
|
||||
#endif
|
||||
|
||||
extern "C" {
|
||||
#include "convolutional_layer.h"
|
||||
|
@ -8,7 +8,9 @@
|
||||
#include <stdio.h>
|
||||
#include <time.h>
|
||||
|
||||
#ifdef CUDNN
|
||||
#pragma comment(lib, "cudnn.lib")
|
||||
#endif
|
||||
|
||||
#ifdef AI2
|
||||
#include "xnor_layer.h"
|
||||
|
Reference in New Issue
Block a user