前提环境:

       显卡更换之前(之前使用的是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

2. 关闭图形界面

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

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

sudo systemctl set-default graphical.target
sudo reboot

3. 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。

4. 打开图形界面并重启

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
声明:本站所有文章,如无特殊说明或标注,均为本站原创发布。任何个人或组织,在未征得本站同意时,禁止复制、盗用、采集、发布本站内容到任何网站、书籍等各类媒体平台。如若本站内容侵犯了原著者的合法权益,可联系我们进行处理。