ubuntu 使用yolov3 yolo-tiny-v3 基于cudnn 7.1 + cuda 9.1 + opencv 3.4.0 以及yolo数据集制作

本文详细介绍了如何在Ubuntu系统上,使用GTX1070显卡搭建YOLOv3和YOLO-tiny-v3的运行环境,包括CUDA 9.1的安装,cuDNN 7.1的下载,OpenCV 3.4.0的源代码安装,以及数据集的制作和训练过程。过程中特别提到了避免驱动问题的方法,以及训练模型时配置文件的调整。
摘要由CSDN通过智能技术生成

1.我在Ubuntu官网上直接下载并安装ubuntu ,显卡使用的gtx1070

2.在nvidia官网家上下载cuda9.1 runtime ubuntu16.04版本

3.下载cudnn7.1 (需要注册)

4.在opencv官网上下载opencv3.4.0 源代码

5.在nvidia官网上下载最新版本的驱动(使用cuda的驱动需要降低内核版本,而我在测试阶段降低内核版本后由于网卡驱动的不兼容断网,所以懒得折腾了就直接使用nvidia官网上的驱动,而如果使用ubuntu addintional driver中的驱动会造成版本的不支持,也懒得折腾索性直接用官网的驱动)

安装过程:1.由于yolo不需要opencv对python的支持,所以不折腾opencv对python的支持安装,直接无脑根据官网的安装方式装,装依赖,然后cmake,make,make install就折腾完了

               2.安装cuda,根据cuda Document中的要求关闭和图形界面有关的所有东西(注:sudo service lightdm stop关闭图形界面),选择的时候不安装驱动,其他的全部yes就行了

               3.安装驱动,一样需要关闭图形界面,省的出奇奇怪怪的问题。sh run就行了。

               4.安装yolo,用github上https://github.com/pjreddie/darknet.git下载源代码,修改Makefile文件opencv,cudnn,cuda=1,然后make就行了。

6.数据集的制作:

6.1 使用ht

评论 1
添加红包

请填写红包祝福语或标题

红包个数最小为10个

红包金额最低5元

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

抵扣说明:

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

余额充值