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,看到以下信息,说明你就已经解决了。

每天进步一点,一起加油哦!关注我,避免迷路。

本文讲述了作者在使用yolov5和人脸识别模型时遇到的GPU问题,由于系统内核升级导致驱动不匹配。通过检查并安装对应版本的DKMS驱动(460.39),作者成功修复了NVIDIA-SMI无法通信的问题,确保了GPU的正常使用。
5万+

被折叠的 条评论
为什么被折叠?



