Windows系统使用conda在虚拟环境中安装CUDA、cuDNN及Pytorch

前言

Anaconda是一个开源的Python发行版本，其中包含了conda、Python等180多个科学包及其依赖项，其中conda可以实现在虚拟环境下安装深度学习所需的环境

Windows系统下不需要在虚拟环境下安装CUDA和cuDNN的可以参考博客：https://blog.csdn.net/Gary_ghw/article/details/108855157

---

安装Anaconda

Anaconda官网下载：https://www.anaconda.com/products/individual#Downloads

 根据所需下载版本，安装过程和普通软件一样（一路Yes就行）

注意：在选在Install之前勾选以上两个选项

---

创建并激活虚拟环境

conda create -n your_env_name python=X.X
如：conda create -n pytorch python=3.7

选择y 回车继续

等待相关库安装完成，激活环境

activate your_env_name
如：activate pytorch

根据提示，使用conda activate pytorch激活环境，但是也可省略conda如上图箭头所示表示进入已进入虚拟环境pytorch

---

安装CUDA和cuDNN

conda install cudatoolkit=10.1 -c https://mirrors.tuna.tsinghua.edu.cn/anaconda/pkgs/free/linux-64/
conda install cudnn=7.6.5 -c https://mirrors.tuna.tsinghua.edu.cn/anaconda/pkgs/main/linux-64/

使用conda在虚拟环境下安装CUDA10.1和cuDNN7.6.5（其中使用了清华源通道下载）

选择y 等待安装完成

---

安装Pytorch

pip install torch==1.7.0+cu101 torchvision==0.8.1+cu101 torchaudio===0.7.0 -f https://download.pytorch.org/whl/torch_stable.html

最后使用pip安装Pytorch1.7.0（安装与CUDA匹配的版本） 完成！！！