方法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/