一、环境配置
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版本
选择自己cuda对应的版本
在yolov5-5.0代码中的 requirements.txt内容进行下载
pip install -i https://pypi.tuna.tsinghua.edu.cn/simple -r requirements.txt
二、源码出处
进入yolov5官网下载yolov5-5.0代码