此博客只提供了一种解决方案,不能保证适用于所有情况。
猜测原因,手动安装的nvidia驱动缺少了某些必要的设置,所以要安装系统推荐的NVIDIA版本。对于已经安装了nvidia+cuda+cudnn+其他依赖Cuda的软件的人来说,只要更改nvidia驱动即可。
一、检查现有显卡
打开终端,输入:
lspci -k | grep -A 2 -i "VGA"
在我的笔记本上显示如下图,如果你的也是类似情形,请继续阅读。
2.nvidia-smi查看当前nvidia工作情况,nvcc -V查看cuda版本,其版本的对应关系参考下图。
3.打开系统设置->软件与更新->附加驱动。可以看到推荐的NVIDIA驱动,如果版本和你已装的CUDA版本兼容那就跳过下一步,如果不兼容。
4.下载与系统推荐NVIDIA驱动兼容的cuda工具包(linux, x86_64, ubuntu, 16.04, runfile(local))
cuda下载地址:https://developer.nvidia.com/cuda-downloads
5.卸载原来的NVIDIA驱动:
用sudo ~/download/NVIDIA-Linux-x86_64-381.09.run --uninstall
进行卸载
6.选中附加驱动中NVIDIA,并点击应用更改,安装系统推荐NVIDIA驱动,重启即可正常扩展。