yolov5环境配置及训练coco128数据集

本文介绍了在配置Yolov5环境时的步骤,包括安装Cuda10.2、cudnn 7.6.5、PyTorch 1.9.0和torchvision 0.10.0。遇到的问题包括错误安装顺序、版本不匹配导致的P R值为0等,并提供了解决方案,如降低Yolov5版本、修改代码和环境变量设置等。此外,还分享了训练COCO128数据集时遇到的多个错误及解决办法。
摘要由CSDN通过智能技术生成

本人小白一个,最近在学习yolov5网络,于是跟着网上的教程配置环境训练等,出现了很多错误,可能会比较乱,先说声抱歉。现在总结一下,算是理清下自己的思路,希望对各位也有些帮助。

环境配置:推荐安装Cuda10.2+cudnn 7.6.5+torch 1.9.0 +torchvision 0.10.0

先按安装orch在安装torchvision,否则会报错

参考连接史上最详细yolov5环境配置搭建+配置所需文件_python_想到好名再改-DevPress官方社区

这个连接中的代码,建议使用yolov5-3.1,我在使用1.0时在训练coco128数据集时,会出现 P R等值为0的情况,根据网上所说的降低版本仍然不行(之前我用的版本是cuda11.3及其相关匹配版本。)

cuda连接:CUDA Toolkit 10.2 Download | NVIDIA Developer

cudnn连接如下:

cuDNN Archive | NVIDIA Developer

这两个连接是官网下载,或者使用下面的命令也可以(应该是,记不太清了)

如果没有找到自己想安装的版本,可以看看下面的连接:https://download.pytorch.org/whl/torch_stable.html

如其中的


表示cuda11.3+python3.8+win64

 下载完,自己所需要的版本后,在自己的环境中pip install +包名 即可,如:

pip install torch-1.10.0+cu102-cp37-cp37m-win_amd64.whl

建议先安装anaconda,创建一个虚拟环境(比如我的是yolov5test),都安装在自己的虚拟环境中。

在conda虚拟环境中安装cuda,cudnn:


  1. conda install cudatoolkit=10.2 -c https://mirrors.tuna.tsinghua.edu.cn/anaconda/pkgs/free/win-64/

  2. conda install cudnn=7.6.5 -c https://mirrors.tuna.tsinghua.edu.cn/anaconda/pkgs/main/win-64/

 激活:安装后打开anaconda prompt,输入activate+ 自己的环境名字 即可,如:activate yolov5test,

在安装cuda时

评论 1
添加红包

请填写红包祝福语或标题

红包个数最小为10个

红包金额最低5元

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

抵扣说明:

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

余额充值