search

hjimce / compress_yolo

对yolo模型进行网络压缩、修剪等实验
Other
50 stars 33 forks source link

编译的时候出现错误 #10

Open smallworld-network-wupeng opened 5 years ago

smallworld-network-wupeng commented 5 years ago

./src/convolutional_layer.c: In function ‘cudnn_convolutional_setup’: ./src/convolutional_layer.c:134:5: error: too few arguments to function ‘cudnnSetConvolution2dDescriptor’ cudnnSetConvolution2dDescriptor(l->convDesc, l->pad, l->pad, l->stride, l->stride, 1, 1, CUDNN_CROSS_CORRELATION); ^~~~~~~ compilation terminated due to -Wfatal-errors. Makefile:84: recipe for target 'obj/convolutional_layer.o' failed make: *** [obj/convolutional_layer.o] Error 1

请问怎么解决呢？

smallworld-network-wupeng commented 5 years ago

关了cudnn后问题解决，但是我原版的darknet开着cudnn也能编译成功的。