windows使用YOLOV5训练模型——搭建编译环境

1:安装Nvidia显卡驱动、cuda和cuDNN

1.1下载安装Navida显卡驱动


NAVIDIA驱动下载地址:https://www.nvidia.cn/Download/index.aspx

 

 下载完成后,在CMD中输入  nvidia-smi 验证是否安装成功

如果有错误

 

2:下载CUDA(本人下载12.2版本)

CUDA下载链接:https://developer.nvidia.com/cuda-toolkit-archive

 

 

3:下载CUDNN

cudnn下载地址(注意cuda版本):https://developer.nvidia.cn/rdp/cudnn-archive

 

 4:安装CUDA

4.1运行cuda,建议使用默认路径

4.2:安装完成后设置环境变量

 计算机属性  -> 系统 -> 系统信息 -> 高级系统设置 -> 系统 的环境变量中可以看到多了CUDA_PATH和CUDA_PATH_V12_2两个环境变量

接下来在系统中添加已下几个环境变量

cuda默认安装路径:C:\Program Files\NVIDIA GPU Computing Toolkit\CUDA\v12.2

CUDA_SDK_PATH:C:\Program Files\NVIDIA GPU Computing Toolkit\CUDA\v12.2
CUDA_LIB_PATH:%CUDA_PATH%\lib\x64
CUDA_BIN_ATH:%CUDA_PATH%\bin
CUDA_SDK_BIN_PATH:%CUDA_SDK_PATH%\bin\win64(12.2版本不存在该文件)
CUDA_SDK_LIB_PATH:%CUDA_SDK_PATH%\common\lib\x64(12.2版本不存在该文件)

 

5:安装CUDNN

将cuCNN解压,将bin、include、lib文件复制到cuda中(C:\Program Files\NVIDIA GPU Computing Toolkit\CUDA\v12.2)

6:cuda安装测试

打开cmd输入:>nvcc --version

即可看到cuda信息

 

7:安装anaconda 

anaconda下载地址:https://www.anaconda.com/download/

Anaconda3-2023.09-0-Windows-x86_64.exe 下载

链接:https://pan.baidu.com/s/1YfzwXMyJ_tT8ti0KcspzEg
提取码:8888

创建Python环境名YOLOv5  python版本3.10

 

conda添加豆瓣源:conda config --add channels http://pypi.douban.com/simple/

conda添加阿里源:conda config --add channels https://mirrors.aliyun.com/pypi/simple/

conda添加清华源: conda config --add channels https://mirrors.tuna.tsinghua.edu.cn/anaconda/pkgs/free

conda添加中科源: conda config --add channels https://mirrors.ustc.edu.cn/anaconda/pkgs/free/

添加后使用命令: conda config --show channels  查看源是否添加成功

 

8:安装yolov5环境

进入环境yolov5,通过命令安装环境

pip install -r requirements.txt

9:下载预训练权重文件

yolov5-6.2 权重文件链接:https://pan.baidu.com/s/1D7Cf98WHEMQJFOjuQJL0lQ
提取码:8888

项目中新建weights文件夹用于存放权重文件

 

 

9:执行测试代码

权重文件下载地址: https://github.com/ultralytics/yolov5/releases/tag/v6.2

https://github.com/ultralytics/yolov5/releases/download/v6.2yolov5n.pt 
https://github.com/ultralytics/yolov5/releases/download/v6.2/yolov5s.pt 
https://github.com/ultralytics/yolov5/releases/download/v6.2lov5m.pt 
https://github.com/ultralytics/yolov5/releases/download/v6.2/yolov5l.pt 
https://github.com/ultralytics/yolov5/releases/download/v6.2/yolov5x.pt 

指定权重文件、设置置信度
python D:\projects\yolov5-6.2\detect.py --weights D:\projects\yolov5-6.2\weights\yolov5s.pt --conf 0.4

识别后的图片会存储在D:\projects\yolov5-6.2\runs\detect 文件夹下

 

 

 
posted @ 2023-12-13 00:51  你看起来真的很好吃  阅读(164)  评论(0编辑  收藏  举报