ubuntu 20.04 安装pytorch、cuda

首先第一步
如果你没有换 pip 源的话
看这里

pip3 install torch torchvision torchaudio

wget https://developer.download.nvidia.com/compute/cuda/repos/ubuntu2004/x86_64/cuda-ubuntu2004.pin
sudo mv cuda-ubuntu2004.pin /etc/apt/preferences.d/cuda-repository-pin-600
sudo apt-key adv --fetch-keys https://developer.download.nvidia.com/compute/cuda/repos/ubuntu2004/x86_64/7fa2af80.pub
sudo add-apt-repository "deb https://developer.download.nvidia.com/compute/cuda/repos/ubuntu2004/x86_64/ /"
sudo apt-get update
sudo apt-get -y install cuda

如果上面报错的话那么使用下面这条命令

sudo apt-get install aptitude
sudo aptitude install cuda

然后选择 Y 继续就安装成功了，安装成功后重启一下大功告成。