一. 问题溯源:更新软件后无法调用
1.事情起因如下,我当时使用的是Ubuntu20.04的
查看系统当前使用内核:
uname -r
发现是:
5.13.0-30-generic
我的问题应该不是显卡驱动的问题,我通过nvidia-smi是可以显示出显卡使用信息的.
二. 重装显卡驱动
2.1 图形化安装(对我不可行)
网上有人通过图形化选择额外驱动对显卡驱动进行安装可以成功,我试了发现不行,输入nvidia-smi结果还是显示找不到驱动,说明这条路走不通.
2.2 PPA源驱动安装(可行)
1.首先查看电脑最适合(推荐)的显卡驱动版本:
我的显示推荐510版本的驱动.
2. 配置PPA:
sudo add-apt-repository ppa:graphics-drivers/ppa
3.更新:
sudo apt-get update
4.安装显卡驱动:
sudo apt install nvidia-driver-510
提示正在安装:
安装块结束会提示你为Secure boot设置密码,如果前面没有禁用secure boot，则会在安装过程中会提示设置一个密码，在重启时需要输入密码验证以禁用secure boot，重启后会出现蓝屏，这时候不能直接选择comnue,而应该按下按键，选择Enroll MOK, 确认后在下一个选项中选择continue,接着输入安装驱动时设置的密码，开机。
建议在BIOS中禁用Secure boot,否则你每次都得输入密码,要不然桌面重启后都会卡进一个图标很大的Ubuntu桌面!
5.安装mesa包:
sudo apt-get install mesa-common-dev
6.完成安装后重启:
sudo reboot
三.重启后显卡驱动确认
1.如果Ubuntu显示器开启夜灯模式,夜灯如果功能正常,初步说明驱动应该是安装上了:
2.输入:
nvidia-smi
发现已经可以找到:
3.测试pytorch-GPU是否正常工作:
import torch
flag = torch.cuda.is_available()
print(flag)
运行后发现:
表明一切回归正常!
四.总结
1.这次还是因为手贱,点击了软件更新,虽然没有更新内核,只更新了几个lib包,结果显卡还是不能正常工作了,来来回回又是改内核,又是重装显卡驱动,最后得出了一个重要结论:如果非必要,不要进行软件更新!
2.当然如果你疯狂地在底线边缘试探,请记得安装如上方法安装显卡驱动,一般显卡驱动重装了就好了!