Ubuntu18.04安装yolov3,darknet并使用显卡的GPU进行加速运算

1:ubuntu18.04.3自带git,可以直接使用git把开源的darknet从github上克隆到本地目录,访问国外的github网站速度很慢,可能会失败,需要多次尝试,输入如下命令:

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

2:下载好了darknet后,切换到darknet目录,并且进行编译

//切换到darknet目录,我这里clone到了home目录下
cd /home/darknet
//编译darknet
sudo make

3:下载训练权重文件,也会比较慢,下载失败需要多试几次

sudo wget https://pjreddie.com/media/files/yolov3.weights

4:darknet下面的data目录有多张自带的用于测试的图片,这里选一张dog.jpg用于测(此时使用的是CPU运算,没有使用GPU进行运算,一般情况在20多秒可以计算完成)

sudo ./darknet detect cfg/yolov3.cfg yolov3.weights data/dog.jpg

结果如下(处理的结果图片默认存储在图形操作界面的/home/darknet目录下的predictions.jpg):

到此就安装好了darknet了,不需要安装显卡;

 

接下来我们需要配置GPU运算,首先需要安装显卡驱动和CUDA,CUDNN,请看我的另外一篇转载的博文:https://blog.csdn.net/u013327224/article/details/107363820

以上显卡的准备工作做好之后修改两个地方的简单配置就可以实现了

1:在darknet目录下修改Makefile文件

//修改前先做个备份
sudo cp Makefile Makefile.bak
//修改文件命令
sudo vim Makefile

如下图所示修改

Ubuntu18.04安装yolov3,darknet并使用显卡的GPU进行加速运算_第1张图片

设置好之后需要重新对darknet进行编译

sudo make

如果运行测试的时候出现out of memory的错误则需要修改cfg目录下的yolov3.cfg文件

sudo vim cfg/yolov3.cfg

Ubuntu18.04安装yolov3,darknet并使用显卡的GPU进行加速运算_第2张图片

使用GPU之后都是毫秒级

Ubuntu18.04安装yolov3,darknet并使用显卡的GPU进行加速运算_第3张图片

你可能感兴趣的:(ubuntu,gpu,ai,神经网络,linux)