系统是ubuntu16.04
首先查看cuda版本
nvidia-smi
从第一行可以看出,显卡驱动版本为410.78,CUDA版本为10.0
以下为cuda版本和显卡驱动的版本对应,可以看出需要更新显卡驱动到418.39及以上的版本
查看cuda和显卡驱动版本对应关系
更新ubuntu驱动版本
$ sudo add-apt-repository ppa:graphics-drivers/ppa
$ sudo apt update
$ ubuntu-drivers devices
这时下面会出现一些选项,选择一个(recommended)。
$ sudo apt install nvidia-driver-430
再次输入 nvidia-smi 查看显卡驱动版本,此时报错:
Failed to initialize NVML: Driver/library version mismatch
然后重启一下就好了。。。。。。。
重启后查看驱动版本和CUDA版本
也不是很清楚CUDA怎么自动更新到10.1了