配置YOLOv5环境并进行训练测试(Win11)

YoloV5训练环境配置

1. 安装、配置CUDA、cuDNN环境

本机配置: win11 + NVIDIA GeForce RTX4050 Laptop GPU+ "CUDA Version: 12.0" + 2.0.0+cu118 + 0.15.1+cu118 + python=3.9.16

IDE:Vscode

CUDA安装:CUDA Toolkit Archive | NVIDIA Developer

cuDNN安装cuDNN Archive | NVIDIA Developer

查看系统支持CUDA版本,打开nvidia控制面板(桌面右键)->选择左下角的系统信息->组件

在这里插入图片描述
在这里插入图片描述

在这里插入图片描述

下载适合自己主机的版本

详细的安装过程参考:CUDA安装教程

查看CUDA是否安装成功,

nvcc --version
OR
nvcc -V

在这里插入图片描述

cuDNN下载完成之后进行解压之后,文件如下所示:
在这里插入图片描述

下载后发现其实cudnn不是一个exe文件,而是一个压缩包,解压后,有三个文件夹,把三个文件夹拷贝到cuda的安装目录下。

CUDA 的安装路径在前面截图中有,或者打开电脑的环境变量查看,默认的安装路径如下:

C:\Program Files\NVIDIA GPU Computing Toolkit\CUDA\v11.0

拷贝时看到,CUDA的安装目录中,有和cuDNN 解压缩后的同名文件夹,这里注意,不需要担心,直接复制即可。cuDNN 解压缩后的同名文件夹中的配置文件会添加到 CUDA安装目录中的同名文件夹中。【此处还是建议还是分别把文件夹的内容复制到对应文件夹中去】

2. 配置Anaconda虚拟环境

详情参考Anaconda 3 安装&虚拟环境配置

在这里插入图片描述

3. YoloV5项目

yoloV5项目:ultralytics/yolov5: YOLOv5 🚀 in PyTorch > ONNX > CoreML > TFLite (github.com)

权重文件地址:yolov5s.pt

下载项目之后,使用pip install -r requirement.txt,下载安装对应的环境

运行detect.py,可以看到runs文件夹下生成检测结果

在这里插入图片描述

在这里插入图片描述

  • 1
    点赞
  • 1
    收藏
    觉得还不错? 一键收藏
  • 0
    评论

“相关推荐”对你有帮助么?

  • 非常没帮助
  • 没帮助
  • 一般
  • 有帮助
  • 非常有帮助
提交
评论
添加红包

请填写红包祝福语或标题

红包个数最小为10个

红包金额最低5元

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

抵扣说明:

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

余额充值