win10+NVIDIA GeForce RTX 3050+torch1.13.1+torchaudio0.13.1+torchvision 0.14.1 +cuda11.7+python3.8+cudnn8.7.0
在环境搭建中踩了许多坑,yolov5环境的搭建需要依赖很多环境,用cpu跑很容易跑单张识别,用GPU跑却踩了很多坑,不过GPU环境搭建起来后续训练数据集我想就会快很对!GPU搭建第一步别着急
1.查看电脑显卡
使用PowerShell
- 使用PowerShell命令:
- 按下
Windows
键,输入PowerShell
,然后右键选择“以管理员身份运行”。 - 在PowerShell窗口中,输入以下命令并按回车:
3.输入后显示我的显卡驱动为3050Get-WmiObject Win32_VideoController | Select-Object Name, Description, DriverVersion
- 同时查看电脑cuda版本可以安装什么版本
-
打开nvidia(桌面右键)->选择左下角的系统信息->组件
第三行,可以看到自己电脑支持的cuda这里显示的是11.7
- 按下
2环境依赖
yolov5依赖环境包含cuda ,cudnn,Anaconda,torch, torchvision,torchaudio,Anaconda下载好了就是powershell命令端,要在这个命令端口搭建虚拟环境
(1)cuda和cudnn安装省略:请参照这个博主的安装方法
CUDA安装教程(超详细)-CSDN博客,他安装的版本为11.0.2,我安装的版本为11.7,解释一下为什么,因为cuda11.0.2对应的pytorch最高版本为1.7,而yolov5算法在跑过程中要求版本是>=1.8.0,所以cuda11.0.2不满足要求。
(2)Anaconda安装省略请参照这个博主Anaconda安装部分
史上最详细yolov5环境配置搭建+配置所需文件_yolov5环境搭建-CSDN博客
安装好后如下所示打开Anaconda powershell Prompt在这个命令端下创建虚拟环境
3安装pytorch
(1)创建虚拟环境
这里的new1为虚拟环境名称,虚拟环境名字随意去,python版本要求3.8
conda create -n new1 python=3.8
(2)激活虚拟环境new1
conda activate new1
(3)安装pytorch
在查阅资料的过程中查到cuda11.7对应的pytorch稳定版本是1.13.1,q请打开下面的网址
download.pytorch.org/whl/torch_stable.html
打开如下所示我下载的版本是cu117/torch-1.13.1%2Bcu117-cp38-cp38-win_amd64.whl
解释一下为什么,因为最开始查询到我的电脑cuda版本是11.7,cu117代表cuda版本,torch-1.13.1%2Bcu117,代表cuda11.7下载的pytorch版本是1.13.1,cp38-cp38代表python版本是3.8,根据cuda版本选择对应的pytorch版本。
pip install C:\Users\PC\Downloads\torch-1.13.1+cu117-cp38-cp38-win_amd64.whl
4.检验pytorch安装好了没有
(1)打开pycharm,将环境切换为new1
点设置--添加解释器--conda环境。。。按照如图设置虚拟环境,这样在Anaconda powershell Prompt中设置的虚拟环境new1就对应到了pycharm中
在设置好后就显示了下图,界面右下角显示new,这里我没有生成new1,仍旧使用之前设置的虚拟new环境,要使用new1就点上一步的new1就好。
同时打开pycharm终端,命令行开始显示new
上述使用创建虚拟环境的命令也可以在这里进行,这里和 Anaconda powershell Prompt是等价的
仍使用以下命令下载pytorch
pip install C:\Users\PC\Downloads\torch-1.13.1+cu117-cp38-cp38-win_amd64.whl
验证pytorch安装好了没有输入以下命令显示true,则说明GPU搭建完成
import torch
torch.cuda.is_available()
4.安装 torchvision,torchaudio
安装torch==1.13.1+cu117
兼容的 torchvision
版本
最初报错!!
ERROR: pip's dependency resolver does not currently take into account all the packages that are installed. This behaviour is the source of the following dependency conflicts. torchvision 0.19.1 requires torch==2.4.1, but you have torch 1.13.1+cu117 which is incompatible.
这个错误表明你的环境中已经安装了一个与 torch==1.13.1+cu117
不兼容的 torchvision
版本(0.19.1),它依赖于 torch==2.4.1
。要解决这个问题,你需要卸载现有的 torchvision
包,然后安装与 torch 1.13.1+cu117
兼容的版本。
以下是如何操作的步骤:
- 卸载现有的
torchvision
包:
shell
pip uninstall torchvision
- 重新安装与
torch 1.13.1+cu117
兼容的torchvision
版本。根据你之前的尝试,这个版本应该是0.14.1+cu117
:
shell
pip install torch==1.13.1+cu117 torchvision==0.14.1+cu117 torchaudio==0.13.1 --extra-inde
这个显示就是下载好了,这里要注意torch, torchvision,torchaudio要相互兼容
5.GPU跑yolov5算法
权重文件为yolov5n6.pt,github下载即可
python .\detect.py --source .\data\images --weights .\weight\yolov5n6.pt
这个就跑好了,保存在 Results saved to runs\detect\exp15