ubuntu22.04更换RTX 4090显卡后,安装驱动和pytorch记录

前提环境:

显卡更换之前(之前使用的是2080ti),已在ubuntu图形界面的”附加驱动”中安装nvidia驱动,并且anaconda环境中的pytorch正常使用。

首先的首先,当然是献上伟大的护舒宝。

ubuntu22.04更换RTX 4090显卡后,安装驱动和pytorch记录
  1. 官网下载4090显卡驱动

Linux x64 (AMD64/EM64T) Display Driver | 520.56 | Linux 64-bit | NVIDIA

注: 确保已在”附加驱动”中安装过nvida驱动,否则请手工禁用 nouveau

  1. 关闭图形界面
sudo systemctl set-default multi-user.target
sudo reboot

对应的打开图形界面命令为(暂时不打开)

sudo systemctl set-default graphical.target
sudo reboot
  1. tty登录之后,安装新版驱动
切换到驱动目录
cd /home/user/download/
权限
sudo chmod a+x NVIDIA-Linux-x86_64-520.56.06.run
安装
sudo ./NVIDIA-Linux-x86_64-520.56.06.run

安装过程中,有个是否安装32位的选项,选否。其余的都是一路yes。

  1. 打开图形界面并重启
sudo systemctl set-default graphical.target
sudo reboot

安装后第一次开机时间较长,耐心等待

经验证,更换4090显卡后,基于cuda10.2编译的pytorch已不受支持;

更换为cuda11.6编译的pytorch可正常使用。

conda install pytorch torchvision torchaudio cudatoolkit=11.6 -c pytorch -c conda-forge

Original: https://blog.csdn.net/u011876981/article/details/127382412
Author: ginogogo
Title: ubuntu22.04更换RTX 4090显卡后,安装驱动和pytorch记录

原创文章受到原创版权保护。转载请注明出处:https://www.johngo689.com/715787/

转载文章受原作者版权保护。转载请注明原作者出处!

(0)

大家都在看

亲爱的 Coder【最近整理,可免费获取】👉 最新必读书单  | 👏 面试题下载  | 🌎 免费的AI知识星球