針對Linux環境需要使用GPU來做AI或Data資源運算時,除須要額外安裝實體顯示卡外,還要安裝驅動,此篇記錄在Dabain環境下安裝NVIDIA驅動與GPU資源監控安裝步驟。

STEP 1:安裝NVIDIA晶片檢查工具

sudo apt install lshw
sudo apt install nvidia-xconfig

sudo lshw -c display

 

STEP 2:啟用contrib、non-free更新來源

sudo apt install software-properties-common


sudo add-apt-repository contrib


sudo add-apt-repository non-free

sudo apt update

 

STEP 3:將nouveau驅動程序列入黑名單

nano /etc/modprobe.d/blacklist-nouveau.conf
放入下列內容
blacklist nouveau
blacklist lbm-nouveau
options nouveau modeset=0
alias nouveau off
alias lbm-nouveau off

 

STEP 4:檢測適合安裝的NVIDIA驅動程式

apt-get install nvidia-detect
檢測完後會顯是適合安裝的指令,再依據建議的指令進行安裝


sudo apt install nvidia-driver


安裝完後重啟主機
reboot

STEP 5:移除顯示卡驅動程式

sudo apt purge nvidia- *
sudo apt autoremove

 

STEP 6:安裝GPU監控工具

apt-get install pip

pip install nvitop
nvitop –monitor auto

參考網站:
———