前提环境:
显卡更换之前(之前使用的是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
文章知识点与官方知识档案匹配,可进一步学习相关知识
Python入门技能树人工智能深度学习238021 人正在系统学习中