深圳幻海软件技术有限公司 欢迎您!

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

2023-02-28

前提环境:    显卡更换之前(之前使用的是2080ti),已在ubuntu图形界面的“附加驱动”中安装nvidia驱动,并且anaconda环境中的pytorch正常使用。   首先的首先,当然是献上伟大的护舒宝。1.官网下载4

前提环境:

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

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

1. 官网下载4090显卡驱动

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

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

2. 关闭图形界面

  1. sudo systemctl set-default multi-user.target
  2. sudo reboot

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

  1. sudo systemctl set-default graphical.target
  2. sudo reboot

3. tty登录之后,安装新版驱动

  1. # 切换到驱动目录
  2. cd /home/user/download/
  3. # 权限
  4. sudo chmod a+x NVIDIA-Linux-x86_64-520.56.06.run
  5. # 安装
  6. sudo ./NVIDIA-Linux-x86_64-520.56.06.run

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

4. 打开图形界面并重启

  1. sudo systemctl set-default graphical.target
  2. sudo reboot

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

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

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

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

文章知识点与官方知识档案匹配,可进一步学习相关知识
Python入门技能树人工智能深度学习238021 人正在系统学习中