anaconda安装pytorch

1,226 阅读1分钟

修改.condarc文件为:

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/cloud/conda-forge/
ssl_verify: true

打开anaconda prompt,依次输入:

conda create -n pytorch python=3.7 

activate pytorch
conda install pytorch
或选择特定版本
cuda search pytorch
conda install pytorch==1.4

如果conda install pytorch报错,则先安装numpy

conda install numpy


安装pytorch 1.4

添加源:

mirrors.tuna.tsinghua.edu.cn/anaconda/cl…

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/pro
  - https://mirrors.tuna.tsinghua.edu.cn/anaconda/cloud/conda-forge/
  - https://mirrors.tuna.tsinghua.edu.cn/anaconda/cloud/pytorch
ssl_verify: true

conda create -n pt1.4 python=3.7 
activate pt1.4
conda install pytorch=1.4

参考资料:

zhuanlan.zhihu.com/p/110332563

官方下载:

conda install pytorch torchvision cudatoolkit=10.1 -c pytorch

要去掉 -c pytorch,这个参数指定了下载通道为官网。