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

最新推荐文章于 2024-08-05 18:41:17 发布

阅读量5.1k

点赞数 2

本文链接：https://blog.csdn.net/weixin_42230524/article/details/80352412

版权

本文详细介绍了如何在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