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 继续就安装成功了，安装成功后重启一下大功告成。

最后

以上就是老实手机为你收集整理的ubuntu 20.04 安装pytorch、cuda的全部内容，希望文章能够帮你解决ubuntu 20.04 安装pytorch、cuda所遇到的程序开发问题。

如果觉得靠谱客网站的内容还不错，欢迎将靠谱客网站推荐给程序员好友。