YOLOV5环境配置

一、环境配置

1.1安装CUDA、CUDNN

我使用的版本为：

cuda: cuda_10.1_win10    cudnn: 7.6.5

版本： cuda_10.1_win10 下载地址：

​​​​​​https://developer.nvidia.com/cuda-90-download-archive?target_os=Windows&target_arch=x86_64

 下载成功之后点击默认安装

版本： cudnn: 7.6.5下载地址（需注册邮箱）： https://developer.nvidia.com/rdp/cudnn-archive

下载完成后解压cudnn压缩包，压缩包中有四个文件，将前三个文件夹当中的内容加载cuda文件夹对应的内容当中。

 1.2选择适配 的pytorch版本

官网：https://pytorch.org/

 

选择自己cuda对应的版本 

 

 

在yolov5-5.0代码中的 requirements.txt内容进行下载

pip install -i https://pypi.tuna.tsinghua.edu.cn/simple -r requirements.txt

二、源码出处

进入yolov5官网下载yolov5-5.0代码

https://github.com/ultralytics/yolov5