默认从conda源下载,安装了一天也没安装上,尝试了几个国内源,最终使用上海交大的源成功安装了。
安装过程如下:
-
从官网(https://pytorch.org/get-started/locally/)上选择自己的相关环境配置,找到对应的
pytorch
安装指令。比如:
conda install pytorch torchvision torchaudio cudatoolkit=10.1 -c pytorch
-
进入自己的conda虚拟环境,首先看下已经安装了哪些镜像源:
conda config --show
或者
直接cat ~/.condarc
如果没有添加好,则添加一下:conda config --add channels https://mirrors.sjtug.sjtu.edu.cn/anaconda/cloud/pytorch/
添加后的~/.condarc
文件内容如下:
channels:
- https://mirrors.sjtug.sjtu.edu.cn/anaconda/cloud/pytorch/
- https://mirrors.sjtug.sjtu.edu.cn/anaconda/pkgs/main/
- https://mirrors.sjtug.sjtu.edu.cn/anaconda/pkgs/free/
- https://mirrors.sjtug.sjtu.edu.cn/anaconda/cloud/conda-forge/
- defaults
- 在命令终端输入指令:
conda install pytorch torchvision torchaudio cudatoolkit=10.1
。
注意一定要去掉官网指令最后面的-c pytorch
,否则还是从conda
源下载。