(个人笔记)yolov4 编译过程及遇到的问题
Ubuntu16.04 已经提前安装好环境 下载
git clone https://github.com/AlexeyAB/darknet.git
进入文件
cd darknet
修改配置文件
vim Makefile
文件中将
GPU=1
CUDNN=1
CUDNN_HALF=1
OPENCV=1
进行修改 然后查看cpu内核一般为8命令为
cat /proc/cpuinfo| grep "cpu cores"| uniq
或者grep 'core id' /proc/cpuinfo | sort -u | wc -l
进行make有几个内核 j 后面就是几
make -j8
编译完成出现这个即为成功 如果出现185: recipe for target ‘obj/network-kernels.o’ failed 需要修改darknet/src/network_kernels.cu这个文件(ubuntu16.04出现这个错误Ubuntu18.04以上没有)
cd src
vim network_kernels.cu
进入vim页面 查找
cudaStreamCaptureModeGlobal
具体命令 :/cudaStreamCaptureModeGlobal
如果之前make过需要进行清除
make clean
然后在
make -j8
提前下好权重文件 命令行输入
./darknet detector test cfg/coco.data cfg/yolov4.cfg yolov4.weights ./data/dog.jpg
远程命令行出现这个为成功
|