1. 前景介绍
前几天还在自己的笔记本上训练了yolov5模型和人脸识别模型,使用的GPU。
然而,几天做项目时,突然发现GPU不能用了,nvidia-smi后报一下错误:
Ubuntu:NVIDIA-SMI has failed because it couldn‘t communicate with the NVIDIA driver.
2. 问题分析与解决
分析问题:这是一个常见问题,经常出现在ubuntu系统中,主要原因还是系统内核升级了,导致新版本内核和原来显卡驱动不匹配。
解决方法的步骤如下:
(1) 查看已安装驱动的版本信息:
ls /usr/src | grep nvidia
结果如下:
我的驱动版本为:460.39
(2) 执行下面两行命令:
sudo apt-get install dkms
sudo dkms install -m nvidia -v 460.39(460.39表示的是驱动版本号)
(3) 执行完上面的命令后,你的显卡就应该能用了。
在终端执行:nvidia-smi,看到以下信息,说明你就已经解决了。
每天进步一点,一起加油哦!关注我,避免迷路。