ubuntu下通过anaconda使用清华镜像安装pytorch
显示已有镜像源
很多时候,下载的镜像源已经无效,我们需要把他先删除。
命令:conda config --show
删除镜像源
命令:conda config --remove channels XXX
XXX指网址
镜像安装pytorch
命令:conda config --add channels https://mirrors.tuna.tsinghua.edu.cn/anaconda/cloud/pytorch/
若向安装特定版本的pytorch,例如1.5.0,cuda10.2的pytorch则运行如下命令:
conda install pytorch=1.5.0 torchvision -c pytorch
注意 -c pytorch去掉,以便快速安装。
测试是否安装成功
命令:如何查看当前pytorch版本:首先进入python环境
python
import torch
print(torch.version__)[version左右两边是双下划线]
查看pytorch的gpu版是否成功
命令:print(torch.cuda.is_avaiable()),若显示True,则表示成功。
如上,安装成功。