文章目录
一、问题提出
深度学习训练模型之时需要GPU支持,但是当我们输入nvidia-smi
之后会提示类似于NVIDIA-SMI has failed because it couldn’t communicate with the NVIDIA driver以及Failed to initialize NVML: Driver/library version mismatch的错误。以下分别进行说明
二、NVIDIA-SMI has failed because it couldn’t communicate with the NVIDIA driver
2.1 安装推荐的nvidia驱动
- 执行命令
sudo ubuntu-drivers devices
会提示推荐安装的nvidia
驱动。安装带有recommended
字样的软件包即可。如下图