【YOLO】darknet安装与测试,服务器,CUDA

darknet安装

git clone https://github.com/pjreddie/darknet
cd darknet
make

这里由于我需要在服务器,gpu运行,先对Makefile文件进行一下修改

GPU=1
CUDNN=1
OPENCV=1

注意千万不要忘了修改nvcc

nvcc=/usr/local/cuda-8.0/bin/nvcc

如果忘了改可能会出现如下错误:/bin/sh: 1: nvcc: not found

改完之后:

这里写图片描述

下面复制这个darknet到服务器上。

scp -r /home/papa/Downloads/darknet [email protected]:/home/ywshang/

这条命令的作用是复制指定文件夹到服务器上。登录服务器make一下吧!

之前做这件事还出过一个错误

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

服务器root权限下运行这句话就ok了~

 sudo ldconfig /usr/local/cuda/lib64

详情请看我的这篇博客:

http://blog.csdn.net/shangpapa3/article/details/76189674

成功现在让我们来测试一下~

wget https://pjreddie.com/media/files/yolo.weights
./darknet detect cfg/yolo.cfg yolo.weights data/dog.jpg

或

./darknet detector test cfg/coco.data cfg/yolo.cfg yolo.weights data/dog.jpg

运行成功

这里写图片描述

这里根据官网步骤简单运行了一下,更多功能设置,请看官网。

https://pjreddie.com/darknet/yolo/

猜你喜欢

转载自blog.csdn.net/shangpapa3/article/details/76273041