[一行命令安装]linux unbuntu20安装cuda, pytorch, torchvision

方法1：使用anaconda

首先在https://pytorch.org/get-started/locally/中抄到这句命令，

复制过去在命令行中直接执行，cuda, pytorch, torchvision都装上啦。

但是还需要最后一步：重新安装显卡驱动。

settings->About->Software Updates->Additional Drivers

安装455就行啦。

然后就可以测试啦。

import torch

torch.cuda.is_available()

如果是True，就说明安装成功啦！

 

方法2：安装cuda，只需要一句：

sudo apt install nvidia-driver nvidia-cuda-dev nvidia-cuda-toolkit nvidia-nsight nvidia-visual-profiler

然后nvcc -V查看，发现安装上了

 但是这个方法可能指定cuda版本不太方便。

 

 另外如果安装中出现/tmp空间不足可以用这两句解决：

sudo mkdir /home/tmp

sudo sh cuda_11.1.0_455.23.05_linux.run --tmpdir=/home/tmp/