1. 程式人生 > >Ubuntu 18.04 安裝 Tensorflow 環境之一:安裝 Nvidia 驅動

Ubuntu 18.04 安裝 Tensorflow 環境之一:安裝 Nvidia 驅動

起因:之前的4個GPU卡的機器經常被別人佔用,隨又向mentor要了一臺機器,這次的是一個單卡 GTX 1080 的伺服器,剛裝了 Ubuntu 18.04 的系統,所以得一步步從頭開始裝。

安裝步驟:

1、刪除舊的Nvidia驅動檔案

sudo apt-get purge nvidia-*

2、加入PPA源

sudo add-apt-repository ppa:graphics-drivers/ppa

3、更新源

sudo apt-get update

4、確認 GPU 卡的型號,檢視推薦的驅動型號

ubuntu-drivers devices

結果如圖:

5、安裝驅動,可自動安裝

sudo ubuntu-drivers autoinstall

也可根據推薦的型號自己選擇安裝

sudo apt install nvidia-410

6、重啟伺服器,有的時候不重啟,相應設定不會生效

sudo reboot

7、測試 nvidia 驅動是否成功安裝

nvidia-smi #如果出現GPU列表,則驅動安裝成功

若成功,會出現以下介面: