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

前提环境：

       显卡更换之前（之前使用的是2080ti），已在ubuntu图形界面的“附加驱动”中安装nvidia驱动，并且anaconda环境中的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