anaconda环境中安装pytorch+cuda+cudnn

基础操作

# 创建新的环境
conda create -n pytorch python=3.6
# To activate this environment, use
#
#     $ conda activate pytorch
#
# To deactivate an active environment, use
#
#     $ conda deactivate

# 删除虚拟环境（env_name是你想要删除的虚拟环境的名字）
 
conda remove -n env_name --all

pytorch安装

在选择清华源安装后，会导致安装成cpu版的pytorch，因此这里选择在官网查看对应的版本，然后在清华源直接下载安装包，在对应的路径下直接安装该安装包。

在anaconda环境中切换到下载的文件夹

conda install --offline pytorch-1.4.0-py3.6_cuda101_cudnn7_0.tar.bz2

conda install --offline torchvision-0.5.0-py36_cu101.tar.bz2

也可以将安装包移动到anaconda路径下的pkgs，然后通过以下指令安装

conda install --use-local pytorch-1.4.0-py3.6_cuda101_cudnn7_0.tar.bz2

conda install --use-local torchvision-0.5.0-py36_cu101.tar.bz2

cuda安装

1. 方法一

conda search cudatoolkit
# 查看可以用来安装的cudatoolkit的版本

conda search cudatoolkit --info
# 查看详细信息

conda install https://mirrors.tuna.tsinghua.edu.cn/anaconda/cloud/conda-forge/win-64/cudatoolkit-10.1.243-h3826478_8.tar.bz2

2. 方法二
   使用官网的命令直接进行安装，这里cuda可以安装成功，但pytorch往往会因为安装过慢无法安装成功，因而选择在清华源下载后安装，如上述pytorch安装所示。
   

cudnn安装

在cudnn仓库中找到适用的版本cuDNN Archive

查看conda仓库中的cudnn版本

conda search cudnn

# 查看详细信息，并根据提供的网址下载即可
conda search cudnn --info

conda install https://mirrors.tuna.tsinghua.edu.cn/anaconda/cloud/conda-forge/win-64/cudnn-8.0.5.39-h36d860d_1.tar.bz2

验证是否安装成功

进入python命令行

1. pytorch

import torch
import torchvision

2. cuda

print(torch.version.cuda)

3. cudnn

print(torch.backends.cudnn.version())

一些问题的解决方法

• 遇到一些报错提示，经查询有numpy和pillow未安装

conda install numpy
conda install pillow

• 解决Solving environment: failed with initial frozen solve. Retrying with flexible solve
  （这里一般不用参考，容易导致安装成cpu版的pytorch，用以上安装方法即可）
  删除之前添加过的源：

conda config --remove-key channels

然后添加正确的清华源：

conda config --add channels https://mirrors.tuna.tsinghua.edu.cn/anaconda/cloud/pytorch/
conda config --add channels https://mirrors.tuna.tsinghua.edu.cn/anaconda/pkgs/free/
conda config --add channels https://mirrors.tuna.tsinghua.edu.cn/anaconda/pkgs/main/
conda config --add channels https://mirrors.bfsu.edu.cn/anaconda/pkgs/free/
conda config --add channels https://mirrors.bfsu.edu.cn/anaconda/pkgs/main/
conda config --add channels https://mirrors.tuna.tsinghua.edu.cn/anaconda/cloud/conda-forge/
conda config --add channels https://mirrors.tuna.tsinghua.edu.cn/anaconda/cloud/msys2/
conda config --add channels https://mirrors.tuna.tsinghua.edu.cn/anaconda/cloud/bioconda/
conda config --add channels https://mirrors.tuna.tsinghua.edu.cn/anaconda/cloud/menpo/
conda config --add channels https://mirrors.tuna.tsinghua.edu.cn/anaconda/cloud/fastai/

conda config --add channels https://mirrors.tuna.tsinghua.edu.cn/anaconda/pkgs/free/
conda config --add channels https://mirrors.tuna.tsinghua.edu.cn/anaconda/pkgs/main/
conda config --set show_channel_urls yes

在用户根目录（C:\Users\用户名）下找到.condarc文件，打开并并编辑，删除其中的default配置行
/ - defaults