如果要编译GPU版的YOLO,需要修改Makefile文件
sudo vim Makefile
GPU=1
CUDNN=1
OPENCV=1
ARCH要根据自己的GPU型号修改
接下来开始编译。
遇到的第一个错误
是cuda的问题
发现没有cuda文件夹,新建个软链接到cuda-10.1,再次编译,解决问题。
第二个错误
这个问题主要原因出现在:cannot find -lcublas
查看了一下发现在/usr/local/cuda/lib64里面没有发现libcublas.so文件,猜到应该是此原因。
于是把anaconda环境中的libcublas.so文件软连接到该目录下
然后重新编译,
成功编译。
执行
./darknet imtest data/eagle.jpg
测试,显示以下。测试成功。