Windows pytorch安装

第一步

安装anaconda
进入这个链接下载好anaconda 的Windows安装包后安装https://docs.anaconda.com/anaconda/install/windows/

第二步

进入Anaconda Powershell prompt后执行

conda config --set show_channel_urls yes

执行后就会在anaconda的安装目录下生成.condarc文件
用以下内容替换

channels:
  - defaults
show_channel_urls: true
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
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

第三步

见链接https://pytorch.org/

gpu版本执行这个
conda install pytorch torchvision cudatoolkit=10.1 -c pytorch
cpu版本执行这个
conda install pytorch torchvision cpuonly -c pytorch

测试

随便去pytorch教程里找一段代码跑一下
http://pytorch123.com/