在自己PC上安装了GPU版本的TensorFlow 1.14,不过由于CUDA是8.0版本的,所以程序运行失败。因此,就需要卸载旧版本的CUDA,并安装CUDA10.0。但是在安装CUDA10.0的过程中,发现驱动版本不匹配,CUDA10.0所需驱动>=410,而我的驱动是384。于是就查阅网上的文档,对驱动进行卸载重装。
-
查看驱动版本
nvidia-smi
-
卸载旧版本
sudo apt-get purge nvidia*
卸载完旧版本之后,我通过“sudo ubuntu-drivers autoinstall”对Nvidia驱动进行了安装,不过安装的默认是最新的430,与我的内核版本不匹配,并提示如下图错误。而且重启之后,屏幕就无法显示图形界面了,于是通过SSH进行配置。
-
输入下条命令,把显卡驱动加入ppa(个人软件包文档,仅支持Ubuntu)
sudo add-apt-repository ppa:graphics-drivers sudo apt-get update
-
输入下条命令,重装410版本驱动
sudo apt-get install nvidia-410 nvidia-settings nvidia-prime
-
重启,并输入下条命令,查看解决方案是否生效
nvidia-smi
-
为防止再次出现这样的问题,可以禁用当前版本的本地更新
sudo apt-mark hold nvidia-410
网友评论