最近复现代码需要用到pytorch1.3.1,查了很多博客,推荐用清华源下载,这个说的很详细https://blog.csdn.net/watermelon1123/article/details/88122020
但是问题是用清华源下载也非常慢,于是我选择删除掉作者推荐的指令中的
红线部分,顺利解决问题~~
sudo gedit ~/.condarc 打开conda的channel。
新发现的问题
torch.cuda.is_available()
False
:
查找之后发现我下载的pytorch是CPU版本的,GPU版本的 有cu的文字。
pytorch pkgs/main::pytorch-1.3.1-cpu_py37h62f~ --> pytorch::pytorch-1.3.1-py3.7_cuda10.1.243_cudnn7.6.3_0
torchvision pkgs/main::torchvision-0.4.2-cpu_py37~ --> pytorch::torchvision-0.4.2-py37_cu101