Ubuntu使用darknet遇到的坑

修改makefile文件

GPU=1
CUDNN=1
NVCC=/usr/local/cuda-9.0/bin/nvcc #即改为本机CUDA安装地址

make后报错

include/darknet.h:16:23: fatal error: cudnn.h: 没有那个文件或目录
compilation terminated.
Makefile:89: recipe for target 'obj/gemm.o' failed
make: *** [obj/gemm.o] Error 1

是因为缺乏头文件cudnn.h

执行

cd /home/zhoujie/.conda/envs/tf/include

sudo cp cudnn.h /usr/include

再次make后报错

/usr/bin/ld: 找不到 -lcudnn
collect2: error: ld returned 1 exit status
Makefile:83: recipe for target 'libdarknet.so' failed
make: *** [libdarknet.so] Error 1

这是因为缺少库文件libcudnn.so

执行

cd /home/zhoujie/.conda/envs/tf/lib

sudo cp libcudnn.so /usr/local/cuda/lib64

make成功

执行推导图片命令后报错

libcudart.so.8.0: cannot open shared object file: No such file or directory

执行

sudo ldconfig /usr/local/cuda/lib64

就可以正常推导和训练