前言
书接上回,今天来配一下Yolov5的环境
准备
介绍一下训练电脑配置和计划装的版本
Windows版本:Windows11
电脑:i9-13900HX+RTX4060
Python版本:3.8
参考博文:
https://blog.csdn.net/m0_51261924/article/details/131650365
https://blog.csdn.net/fly905/article/details/136971768
要更新英伟达显卡驱动的话自己去官网下载就行
查看CUDA版本:CMD输入
nvidia-smi
安装Miniconda以及Pycharm
https://docs.anaconda.com/free/miniconda/
https://www.jetbrains.com.cn/pycharm/
傻瓜式安装即可,Miniconda的安装记得勾选添加进Path环境变量选项
创建yolov5虚拟环境,下载依赖项
在应用程序里找到Anaconda Prompt
输入
conda create -n yolov5 python=3.8
回车后等待一会,把依赖项下载好
下载yolov5工程,权重文件
跟文章绑定的资料有,下载要钱的话私信发
下载C++生成工具
安装pycocotools时需要C++的环境所以需要安装这个
微软C++官网下载
点击下载生成工具
傻瓜式安装,安装完成后打开,下载通用windows平台生成工具即可
Pycharm工程配置,编译器设置以及Python包下载
解压之后正常的工程结构是这样
建议把权重文件(pt后缀)也复制到工程目录
右下角切换成刚才在Anaconda Prompt中创建的虚拟环境
一般是需要添加的(添加部分跟着蓝色高亮条操作)
找到自己虚拟环境的python.exe的路径即可
打开detect.py后会发现缺很多包,点击install requirement即可
pip install matplotlib==3.2.2
pip install numpy==1.18.5
pip install opencv-python==4.1.2//这个有问题,在提示的版本里挑一个相对较近的版本安装
pip install Pillow==5.3.0
pip install PyYAML==5.3.1
pip install scipy==1.4.1
pip install tqdm==4.41.0
pip install tensorboard==2.4.1
pip install seaborn==0.11.0
pip install pandas
pip install thop
pip install pycocotools>=2.0
常见问题:
1、protobuf版本高于3.9.2
解决办法:
找到这个包,下载指定版本3.9.2
2、torch版本和torchvision版本对不上(这个问题挺难搞)
pip install --upgrade torch==2.2.0 torchvision -i https://pypi.tuna.tsinghua.edu.cn/simple/
目前我就遇到这两个问题,修完之后点击运行就可以在runs\detect\exp2中找到结果
到此,PC端配置yolov5环境,结束,如果有说的不对或者不完善的欢迎大佬批评指正