Windows环境下使用conda安装pytorch

1、首先安装anaconda3（Python3.7以上），并查看Python版本。

  

2、单击启动Anaconda Prompt，去创建pytorch的虚拟环境：

conda create --name pytorch python=3.7.6

会遇到选择 Proceed([y]/n)? ，输入y回车即可。

3、进入创建的pytorch虚拟环境

conda activate pytorch

  

4、准备工作完成，Anaconda prompt窗口不要关闭，最小化即可，接下来先查看自身电脑是否支持GPU，再决定安装pytorch。

        查看电脑GPU驱动版本，如果Driver version的数值大于390即可选择电脑带GPU的选项。如果cmd不支持nvidia-smi命令，请在系统环境变量path中把C:\Program Files\NVIDIA Corporation\NVSMI加进去即可。

  

5、进入pytorch官网找到对应的下载语句，主要区分于你的电脑是否有GPU。像我的就基本不具备GPU，就直接安装CPU版本的。

 6、国内一般使用清华镜像源进行安装，所以接下来是修改镜像路径。回到Anaconda prompt窗口，并在虚拟环境中输入如下命令：

conda config --set show_channel_urls yes

   

此时在用户目录下（一般是当前目录下)有一个.condarc文件，记事本打开后将以下内容贴入并保存：

channels:
  - https://mirrors.tuna.tsinghua.edu.cn/anaconda/cloud/pytorch/win-64/
  - defaults
show_channel_urls: true
channel_alias: https://mirrors.tuna.tsinghua.edu.cn/anaconda
default_channels:
  - https://mirrors.tuna.tsinghua.edu.cn/anaconda/pkgs/main
  - https://mirrors.tuna.tsinghua.edu.cn/anaconda/pkgs/free
  - https://mirrors.tuna.tsinghua.edu.cn/anaconda/pkgs/r
  - https://mirrors.tuna.tsinghua.edu.cn/anaconda/pkgs/pro
  - https://mirrors.tuna.tsinghua.edu.cn/anaconda/pkgs/msys2
custom_channels:
  conda-forge: https://mirrors.tuna.tsinghua.edu.cn/anaconda/cloud
  msys2: https://mirrors.tuna.tsinghua.edu.cn/anaconda/cloud
  bioconda: https://mirrors.tuna.tsinghua.edu.cn/anaconda/cloud
  menpo: https://mirrors.tuna.tsinghua.edu.cn/anaconda/cloud
  pytorch: https://mirrors.tuna.tsinghua.edu.cn/anaconda/cloud
  simpleitk: https://mirrors.tuna.tsinghua.edu.cn/anaconda/cloud

 7、将第5步的安装语句拷贝到终端执行。

在开始菜单中，打开 Anaconda Prompt，查看最左边括号中是 base 还是 pytorch。如果是 [base]，使用 conda activate pytorch 进入 pytorch 环境中。之后粘贴回测即可。

conda install pytorch torchvision torchaudio cpuonly -c pytorch

如上述语句安装卡顿，请去掉 -c pytorch:

其中，-c pytorch参数指定了conda获取pytorch的channel,在此指定为conda自带的pytorch仓库。因此，只需要将-c pytorch语句去掉，就可以使用清华镜像源快速安装pytorch了。 

8、验证是否安装成功

（1）在命令行左边为 pytorch 环境中，输入 python

（2）之后，输入 import torch，如果没有报错，意味着 PyTorch 已经顺利安装了