眾所周知,Ubuntu系統自帶的顯示卡驅動是開源的顯示卡驅動。當我們在配置Ubuntu深度學習工作站時,通常需要將這個開源的顯示卡驅動更換為NVIDIA的官方驅動。
而根據工作站的顯示卡配置以及專案所需的CUDA版本,我們也需要切換到對應版本的顯示卡驅動。然而在Ubuntu預設的軟體與更新
裡一般只能找到很少的幾個NVIDIA專有驅動,這就導致我們可能沒法順利配置我們想要的深度學習開發環境。
為了解決這個問題,這裡就講一下如何在軟體與更新
當中配置第三方倉庫PPA,並且通過第三方倉庫安裝NVIDIA顯示卡驅動。
一、使用國內軟體源加速
在原生Ubuntu系統中執行sudo apt-get update
或者sudo apt-get upgrade
等命令時,往往命令執行速度是比較慢的。原因在於許多依賴包都是直接通過國外的源下載的,這裡可以手動設定國內的映象,步驟如下:
1、開啟sources.list這個檔案
sudo gedit /etc/apt/sources.list
2、清空裡面的內容,把以下的源新增進去
# 阿里雲源
deb http://mirrors.aliyun.com/ubuntu/ bionic main restricted universe multiverse
deb http://mirrors.aliyun.com/ubuntu/ bionic-security main restricted universe multiverse
deb http://mirrors.aliyun.com/ubuntu/ bionic-updates main restricted universe multiverse
deb http://mirrors.aliyun.com/ubuntu/ bionic-backports main restricted universe multiverse
##測試版源
deb http://mirrors.aliyun.com/ubuntu/ bionic-proposed main restricted universe multiverse
# 原始碼
deb-src http://mirrors.aliyun.com/ubuntu/ bionic main restricted universe multiverse
deb-src http://mirrors.aliyun.com/ubuntu/ bionic-security main restricted universe multiverse
deb-src http://mirrors.aliyun.com/ubuntu/ bionic-updates main restricted universe multiverse
deb-src http://mirrors.aliyun.com/ubuntu/ bionic-backports main restricted universe multiverse
##測試版源
deb-src http://mirrors.aliyun.com/ubuntu/ bionic-proposed main restricted universe multiverse
# 清華大學源
deb http://mirrors.tuna.tsinghua.edu.cn/ubuntu/ bionic main restricted universe multiverse
deb http://mirrors.tuna.tsinghua.edu.cn/ubuntu/ bionic-security main restricted universe multiverse
deb http://mirrors.tuna.tsinghua.edu.cn/ubuntu/ bionic-updates main restricted universe multiverse
deb http://mirrors.tuna.tsinghua.edu.cn/ubuntu/ bionic-backports main restricted universe multiverse
##測試版源
deb http://mirrors.tuna.tsinghua.edu.cn/ubuntu/ bionic-proposed main restricted universe multiverse
# 原始碼
deb-src http://mirrors.tuna.tsinghua.edu.cn/ubuntu/ bionic main restricted universe multiverse
deb-src http://mirrors.tuna.tsinghua.edu.cn/ubuntu/ bionic-security main restricted universe multiverse
deb-src http://mirrors.tuna.tsinghua.edu.cn/ubuntu/ bionic-updates main restricted universe multiverse
deb-src http://mirrors.tuna.tsinghua.edu.cn/ubuntu/ bionic-backports main restricted universe multiverse
##測試版源
deb-src http://mirrors.tuna.tsinghua.edu.cn/ubuntu/ bionic-proposed main restricted universe multiverse
3、更新軟體列表
新增好國內軟體源之後,執行命令更新軟體列表:
sudo apt-get update
新增第三方倉庫PPA
步驟如下:
1、執行以下命令新增顯示卡驅動的第三方倉庫:
sudo add-apt-repository ppa:graphics-drivers/ppa
2、更新軟體列表
sudo apt update
二、禁用開源驅動nouveau
前面提到,Ubuntu預設是使用開源驅動nouveau的。因此在安裝NVIDIA驅動前需要禁用nouveau,方法如下:
1、連續執行以下兩條命令
sudo bash -c "echo blacklist nouveau > /etc/modprobe.d/blacklist-nvidia-nouveau.conf"
sudo bash -c "echo options nouveau modeset=0 >> /etc/modprobe.d/blacklist-nvidia-nouveau.conf"
這兩條命令將在檔案/etc/modprobe.d/blacklist-nvidia-nouveau.conf
中寫入禁用nouveau的配置,當然也可以直接找到這個檔案用編輯器(vim或者gedit)把配置寫進去。
2、然後可以執行以下命令驗證以上的指令輸入到/etc/modprobe.d/blacklist-nvidia-nouveau.conf
這個檔案中:
cat /etc/modprobe.d/blacklist-nvidia-nouveau.conf
如果顯示以下資訊就表示已經成功禁用了:
blacklist nouveau
options nouveau modeset=0
三、正式安裝NIVDIA驅動
1、解除安裝舊的NVIDIA驅動(如果需要的話):
sudo apt-get purge nvidia*
2、把顯示卡驅動加進PPA
sudo add-apt-repository ppa:graphics-drivers
sudo apt-get update
3、查詢最新的NVIDIA驅動
sudo apt-cache search nvidia
4、通過以下指令檢視推薦的驅動版本
ubuntu-drivers devices
5、開啟系統裡的Ubuntu軟體
,然後點選左上角的“軟體和更新”,點選“附加驅動”,選擇其中一個專有驅動,然後點選進行“應用更改”。安裝完之後系統會要求使用者進行重啟,這裡按指示執行重啟操作即可。
6、重啟成功後可以查詢設定
裡的詳細資訊
,驅動安裝正確的話應該能看到NVIDIA顯示卡的相關資訊。