Pytorch及Yolov5环境配置及踩坑
阅读原文时间:2023年07月09日阅读:1

1.何为Yolov5

yolo是计算机视觉方面用来进行目标检测的一个十分出名的开源框架,我搜不到官方的对此概括性的定义,但实际上也没什么必要,更重要的是会使用,更更重要的是理解其底层的实现逻辑

知乎

知乎2

对于我现在的数学水平,想了解底层实现逻辑的10%,都有些天方夜谭了,之后我可能会开一个文章专门跟进我学习的进度,也可能会在这篇文章的下面更新学习流程

1.何为卷积?

视频:【工程数学基础】3_变声的基础原理_理解卷积的含义_线性时不变系统的冲激响应与卷积

2.环境配置

1.Python

其实也可以在C++下进行Yolo的配置,但是我不想费事了,直接就在Python下搞了,而且我也推荐人们在Python下使用Yolo,虽然效率确实没有C++高,但实际上都是调用API,也没有什么差别

想要配置yolo这个大家伙,需要分为几个步骤

CUDA、CUDNN、显卡驱动

首先说一下安装这三个东西的原因,如果我们要进行物品识别,那就肯定要对视频进行一个分析,我们有两条路可以选,一条是不用你的GPU进行加速,另一条是用你的GPU进行加速,有什么区别呢?如果你不用GPU加速,那么你的CPU就会很难受,因为完全在用CPU跑训练,但是CPU根本就不是用来图形渲染的,你这不是难为人家吗?况且Pytorch辛辛苦苦写了这么多GPU加速,你视而不见,有点可惜了不是吗?

所以我们必须要搞一手GPU加速,为了我们好,也为了电脑多活几天

登录英伟达官网,选择相应的显卡型号,安装显卡驱动即可,验证安装是否成功的方式是在cmd中输入nvidia smi

CUDA的安装也是一样的,在英伟达官方根据你的显卡型号选择对应的CUDA版本即可,一般不是安培架构的GPU的话,推荐还是选择10.2这个版本,图灵架构在这个版本下会有更优越的表现

至于CUDNN,这个东西有点麻烦,你需要注册一个英伟达的账号才可以在官网下载

Anaconda

Anaconda指的是一个开源的Python发行版本,其包含了conda、Python等180多个科学包及其依赖项。

Anaconda的安装也没有问题,在官网搜索下载,注意在安装程序时加入系统变量即可了

Pytorch

Pytorch的安装也没问题,搜索官网,唯一需要注意的是选择与你的CUDA版本所相对应的Pytorch版本安装即可,至于在哪里安装。

首先打开Anaconda prompt,输入anaconda create --name <env_name>,这里的env_name是你要创建的环境的名字

之后输入anaconda activate <env_name>进入你所要安装至的环境中,在Pytorch官网找到与你的CUDA版本所相对应的Pytorch安装指令,输入即可

用过Linux的应该都知道的,由于网络原因,下载之前要先进行一个换源的操作,推荐换清华源来进行安装,这个百度即可

Paddle

终于到了最后一步,我们需要安装百度的一个深度学习平台Paddle

同Pytorch的安装方法一样,进行你的Pytorch环境,在Paddle官网选择与你的CUDA版本相对应的Paddle版本安装指令进行安装即可,这里使用的是百度源,下载速度还是可以的


看似安装过程没什么问题,那么到底是哪里可能出问题呢?

我的问题出在环境安装结束后,缺少了很多其他第三方库的安装,例如torch进度条可视化,Scipy库等等,解决方法很简单,我们在anaconda下进入对应的pytorch环境,利用pip安装即可,注意换源的问题

之后又遇到了CUDA版本或是Torchvision与Torch不对应的问题,这个问题要到Pytorch在github上的开源仓库中进行解决,不幸的是我们只能先卸载我们辛苦安装的torch,之后找到与CUDA版本对应的pytorch进行安装即可

最后遇到的问题是模型训练时无法调用CUDA进行GPU加速,但实际上我是已经安装了CUDA的,这个问题的出现是因为没有安装CUDNN,去官网下载CUDNN添加至系统变量后重启电脑即可

这么大一个东西配置下来总而言之就是玄学,多踩坑也是学习了……