PyTorch安装

官方安装

打算获取最新的PyTorch版本可以直接访问PyTorch官网.
直接提供最新版本的多平台多环境下的安装指令.

如果希望安装旧版PyTorch可以查看版本记录.
注: PyTorch官网访问和下载速度可能非常缓慢, 可能需要用额外的方式访问.

Conda安装

直接在conda环境执行如下语句:

conda install pytorch=0.4.1 cuda92 -c pytorch

但速度可能会非常缓慢甚至频繁失败, 如果多次失败无法正常安装可以考虑换源.

Conda换源

可以前往清华源Conda镜像查看更新.
一般来说可以在文件夹下找到.condarc文件并添加如下语句(未来可能有所变化请以官方为准):

channels:
  - 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

Win用户可能默认没有.condarc文件, 可以先执行如下语句conda config --set show_channel_urls yes再去用户目录查看.

清华源安装

如果因为各种原因希望手动安装PyTorch, 可以先到清华源的PyTorch镜像下载对应版本的PyTorch安装文件.
随后执行conda install --use-local C:\ProgramData\Anaconda3\pkgs\pytorch-0.4.1-py37_cuda92_cudnn7he774522_1.tar.bz2进行安装, 文件具体路径可能会有所改变.

安装成功后, 强烈建议在环境变量系统变量PATH中添加C:\Program Files\NVIDIA Corporation\NVSMI以防止各种玄学错误, 添加后记得重启.