YOLOv5要求:Python》=3.7 pytorch》=1.7
1.驱动下载
下载旧版本英伟达驱动
关于何如在英伟达官网上下载历史驱动的方法_BeatifulBoyBoy的博客-CSDN博客_英伟达历史驱动
我选择的是419.17版本
2.CUDA及cudnn 安装
根据驱动版本(419.17)找到控制面板,找到组件查看cuda版本(我的是10.1),所以选择安装10.1版本。
点击进入CUDA Toolkit Archive网址,下载其中的cuda10.1版本。
cudnn根据cuda安装对应的版本,将解压的三个文件复制到cuda安装的文件中。
深度学习环境搭建(GPU)CUDA安装(完全版)_小邢同学的博客-CSDN博客_cuda搭建
3.anaconda安装
下载Anaconda3-2020.02-Windows-x86_64
选择自动安装环境变量
配置清华镜像源
Anaconda配置清华镜像源_克豪的博客-CSDN博客_anaconda 清华镜像
4.安装pytorch
首先配置虚拟环境
# 创建
conda create -n env_name python=x.x
我选择的Python3.7
然后进入pytorch官网。点击进入Pytorch官网
选择之前的版本
安装pytorch1.7.1,cuda选择对应的10.1版本
在自己创建的虚拟环境中输入即可安装成功。
conda install pytorch==1.7.1 torchvision==0.8.2 torchaudio==0.7.2 cudatoolkit=10.1 -c pytorch