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

如下图所示修改

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

sudo make

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

sudo vim cfg/yolov3.cfg

使用GPU之后都是毫秒级

  • 1
    点赞
  • 19
    收藏
    觉得还不错? 一键收藏
  • 1
    评论

“相关推荐”对你有帮助么?

  • 非常没帮助
  • 没帮助
  • 一般
  • 有帮助
  • 非常有帮助
提交
评论 1
添加红包

请填写红包祝福语或标题

红包个数最小为10个

红包金额最低5元

当前余额3.43前往充值 >
需支付:10.00
成就一亿技术人!
领取后你会自动成为博主和红包主的粉丝 规则
hope_wisdom
发出的红包
实付
使用余额支付
点击重新获取
扫码支付
钱包余额 0

抵扣说明:

1.余额是钱包充值的虚拟货币,按照1:1的比例进行支付金额的抵扣。
2.余额无法直接购买下载,可以购买VIP、付费专栏及课程。

余额充值