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