ubuntu下通过anaconda使用清华镜像安装pytorch

时间:2024-03-28 15:43:38

显示已有镜像源

很多时候,下载的镜像源已经无效,我们需要把他先删除。
命令: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,则表示成功。
ubuntu下通过anaconda使用清华镜像安装pytorch如上,安装成功。