眾所周知,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顯卡的相關信息。