Pytorch的安装方法总结

1.环境创建

打开anaconda prompt，并创建新的环境：

conda create -n pytorch python=3.6

这里不详细解释为什么是3.6，因为这里面存在cuda，cudnn的适配问题。

点击y确认安装表，然后安装完成后输入：

activate pytorch

进入该环境中

2.安装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/cloud/conda-forge 
conda config --add channels https://mirrors.tuna.tsinghua.edu.cn/anaconda/cloud/msys2/

然后进入pytorch的官网，根据自己的配置选择对应的属性，网站会自动反馈你的安装指令，如下图所示：

复制前段指令：conda install pytorch torchvision torchaudio cudatoolkit=10.1

然后静待安装就好了

3.验证安装

输入以下代码即可验证

import torch    # 如正常则静默
a = torch.Tensor([1.])    # 如正常则静默
a.cuda()    # 如正常则返回"tensor([ 1.], device='cuda:0')"
from torch.backends import cudnn # 如正常则静默
cudnn.is_acceptable(a.cuda())    # 如正常则返回 "True"