百度工控机安装GPU驱动程序出现Failed to initialize NVML: Driver/library version mismatch

问题描述

按照百度官方教程安装驱动程序，在终端输入nvidia-smi后出现Failed to initialize NVML: Driver/library version mismatch。

问题根源

NVIDIA 内核驱动版本与系统驱动不一致。
输入以下命令可以分别得到内核驱动和系统驱动的版本。

cat /proc/driver/nvidia/version
cat /var/log/dpkg.log | grep nvidia

最后发现内核驱动版本是430.50，而系统驱动版本是430.64。

解决方法

将脚本中的驱动的版本号修改成430.64。

关于sudo运行脚本的方法

第一次在ubuntu中使用脚本下载，方法是先让脚本可执行，然后sudo执行。

chmod +x install-nvidia.sh
sudo ./install-nvidia.sh