1. 程式人生 > 實用技巧 >一. 使用CUDA中的NVIDIA驅動選項安裝NVIDIA驅動

一. 使用CUDA中的NVIDIA驅動選項安裝NVIDIA驅動

CUDA和NVIDIA顯示卡驅動安裝

cuda的安裝選項中其實包含了nvidia驅動的安裝選項,不過網上好多資料都說不要再cuda中勾選nvidia驅動,而要自己去nvidia官網自己查好型號下載安裝檔案,手動安裝nvidia驅動。
其實主要都是擔心不同版本相容性問題,我這裡忍不住還是想嘗試一下用cuda中的nvidia驅動會出現什麼問題。

檢查是否已經安裝的nvidia驅動:

nvidia-smi
執行這條命令,如果有以下資訊說明已經有nvidia驅動了,結束。
如果沒有以下資訊,說明還沒有nvidia驅動,需要安裝。
在這裡插入圖片描述

禁用nouveau

安裝Nvidia顯示卡驅動和系統自帶的nouveau驅動會有衝突,所以要禁用nouveau。

lsmod | grep nouveau
執行這條命令,如果有資訊,說明nouveau還在執行著,要禁用。
禁用方法:
sudo vi /etc/modprobe.d/blacklist.conf
在文字最後新增如下內容:
blacklist nouveau
options nouveau modeset=0
更新一下:
sudo update-initramfs -u
重啟:
reboot
再次檢視是否已經禁用:
lsmod | grep nouveau
執行這條命令,發現沒有任何資訊輸出了,說明已經禁用了。

安裝gcc

sudo apt install build-essential

刪除舊的nvidia驅動

sudo apt-get remove nvidia-*
sudo apt-get autoremove

或者:
sudo apt-get --purge remove nvidia-*
sudo ./NVIDIA-Linux-x86_64-410.57.run -uninstall
sudo update-initramfs -u
sudo reboot now

更新倉庫

sudo apt-get update

檢視系統推薦的nvidia驅動版本

ubuntu-drivers devices
可以看到recommended的是nvidia-driver-450-server
這裡可以我沒有直接用命令(sudo apt-get install nvidia-settings nvidia-driver-430 nvidia-prime 或者 sudo ubuntu-drivers autoinstall)安裝該驅動,因為看別人才那麼多坑,總覺得事情沒這麼簡單,哈哈,而且我也想試試用cuda中的nvidia驅動安裝選項來安裝驅動,所以這條命令我就看看而已。

在這裡插入圖片描述

cuda安裝成功

cuda官網
在這裡插入圖片描述
執行xxxxxxx.run的過程中,會提示nvidia driver、cuda toolkit相關的資訊,就直接選"Install"繼續安裝,就是這裡將進行nvidia驅動的自動匹配版本和安裝。
如下所示,這樣就表示cuda安裝完成了:
在這裡插入圖片描述
根據上述提示資訊,需要配置環境變數:
sudo vi ~/.bashrc
開啟文字後,在最後新增如下資訊,具體因人而異,要和上圖紅框中提示資訊一致:
export PATH=/usr/local/cuda-11.1/binKaTeX parse error: Expected '}', got 'EOF' at end of input: {PATH:+:{PATH}}
export LD_LIBRARY_PATH=/usr/local/cuda-11.1/lib64${LD_LIBRARY_PATH:+:${LD_LIBRARY_PATH}
在這裡插入圖片描述
使其生效:
source ~/.bashrc

驗證cuda是否成功安裝:
cd /usr/local/cuda-10.1/samples/1_Utilities/deviceQuery
sudo make
./deviceQuery
結果如下說明成功。
在這裡插入圖片描述

nvidia驅動安裝成功

按照商住步驟完成後,再次檢視nvidia-smi資訊,可以看到已經安裝成功。
在這裡插入圖片描述