1.介绍
前不久tensorflow出来2.0版本,当我们去尝试tf2.0的时候,可能会遇到一个问题,就是需要安装cuda10的版本,我们才能正确安装tf2.0,而此时的显卡驱动版本太低,比如下图中是384.54的版本,只支持cuda9,因此,我们不得不升级驱动。本人在网上看了好几个教程,稀里糊涂的总结出了一套简单方案,此方案目前只在远程服务器上尝试过。
2.步骤
2.0 卸载原始显卡驱动(如果没有安装显卡驱动,跳过即可)
执行:
sudo apt-get remove nvidia*
2.1 禁止集成的nouveau驱动
Ubuntu系统集成的显卡驱动程序是nouveau,它是第三方为NVIDIA开发的开源驱动,我们需要先将其屏蔽才能安装NVIDIA官方驱动。 如果我们直接安装驱动的话,往往会报错:ERROR: The Nouveau kernel driver is currently in use by your system. This driver is incompatible with the NVIDIA driver。
将驱动添加到黑名单blacklist.conf中,但是由于该文件的属性不允许修改。所以需要先修改文件属性。
查看属性:
sudo ls -lh /etc/modprobe.d/blacklist.conf
修改属性:
sudo chmod 666 /etc/modprobe.d/blacklist.conf
用vim编辑器打开:
sudo vim /etc/modprobe.d/blacklist.conf
在该文件后添加以下几行:
blacklist vga16fb
blacklist nouveau
blacklist rivafb
blacklist rivatv
blacklist nvidiafb
并执行:
sudo update-initramfs -u
重启(reboot)后执行:
lsmod | grep nouveau
2.2 禁用X服务
执行:
sudo /etc/init.d/lightdm stop
2.3 安装NVIDIA显卡驱动
安装显卡驱动部分刚开始按照网上教程关这关那的入了不少的坑,后来发现用ppa源安装显卡驱动的方式是最简单的。
第一步:终端下使用如下命令添加Graphic Drivers PPA
sudo add-apt-repository ppa:graphics-drivers/ppa
第二步:终端下运行如下命令
sudo apt-get update
第三步:终端下运行如下命令可以查看到可供安装的显卡驱动版本
ubuntu-drivers devices
第四步:选好要安装的驱动版本,运行如下代码安装驱动
sudo apt-get install nvidia-driver-xxx
重启即可
reboot