在笔者之前的文章中介绍了如何在ubuntu服务器上安装conda的方法。这篇文章主要介绍一下如何继续安装gpu版本的pytorch,以及遇到http error导致一直安装失败的解决方法。
切换虚拟环境
conda activate xxxx
修改默认下载源
vim ~/.condarc
输入a切换成输入模式
删除所有源包括default(非常重要,笔者在这里卡了三天没成功就是因为没删除其他源和default!!!!!)
添加
conda config --add channels http://mirrors.tuna.tsinghua.edu.cn/anaconda/cloud/pytorch/
conda config --add channels http://mirrors.tuna.tsinghua.edu.cn/anaconda/pkgs/free/
conda config --add channels http://mirrors.tuna.tsinghua.edu.cn/anaconda/pkgs/main/
输入esc退出,输入:wq保存并退出
在当前环境下输入(可选)
conda config --set show_channel_urls yes
下载gpu torch
去torch官网找到对应的下载命令,我的如下
conda install torch=版本号 torchvision=版本号 cudatoolkit=版本号
注意千万不要加-c pytorch!!!!!
等待安装成功
python
import torch
print(torch.cuda.is_available())
当输出True则安装成功