查看顯卡型號
不知道顯卡型號的可以通過此命令查看,但可能有的新顯卡無法識別。
lspci | grep VGA
下載顯卡驅動程序
在 NVIDIA 官網 或 GeForce 官網 下載所需的顯卡驅動程序。
需要注意的是顯卡驅動需要和 CUDA 版本對應,而 CUDA 版本又要和 PyTorch 或 TensorFlow 的版本對應,所以原則上是越新的版本越好,因為可以支持更多版本的深度學習框架。
禁用 nouveau 驅動
1.使用下述命令可以查看 nouveau 驅動是否運行: lsmod | grep nouveau 若出現下述結果: nouveau 1863680 9 video 49152 1 nouveau ttm 102400 1 nouveau mxm_wmi 16384 1 nouveau drm_kms_helper 180224 1 nouveau drm 479232 12 drm_kms_helper,ttm,nouveau i2c_algo_bit 16384 2 igb,nouveau wmi 28672 4 intel_wmi_thunderbolt,wmi_bmof,mxm_wmi,nouveau 說明 nouveau 驅動正在運行。 2.運行下述命令禁用該驅動: 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" 檢查命令是否正確: cat /etc/modprobe.d/blacklist-nvidia-nouveau.conf 若出現下述結果說明命令正確: blacklist nouveau options nouveau modeset=0 3.更新設置並重啟: sudo update-initramfs -u sudo reboot 4.重啟后重新輸入下述命令: lsmod | grep nouveau 若沒有任何輸出說明禁用 nouveau 驅動成功
驅動路徑:/home/wonxing/down
NVIDIA-Linux-x86_64-450.80.02.run
安裝 NVIDIA 顯卡驅動
1.安裝依賴:
sudo apt install gcc g++ make
2.登錄時按 ctrl + alt + F2
進入命令行並使用用戶名密碼登錄,並輸入 sudo telinit 3
打開一個新的 TTY1 界面。如果是 SSH 遠程連接,則不需要做上述步驟。
3.安裝驅動:
cd down
sudo ./NVIDIA-Linux-x86_64-450.80.02.run
並按下述選項選擇:
4.安裝成功后輸入 nvidia-smi
,若有類似下述輸出證明顯卡安裝成功:
+-----------------------------------------------------------------------------+ | NVIDIA-SMI 418.56 Driver Version: 418.56 CUDA Version: 10.1 | |-------------------------------+----------------------+----------------------+ | GPU Name Persistence-M| Bus-Id Disp.A | Volatile Uncorr. ECC | | Fan Temp Perf Pwr:Usage/Cap| Memory-Usage | GPU-Util Compute M. | |===============================+======================+======================| | 0 GeForce RTX 208... Off | 00000000:19:00.0 Off | N/A | | 52% 57C P0 59W / 250W | 0MiB / 10989MiB | 0% Default | +-------------------------------+----------------------+----------------------+ | 1 GeForce RTX 208... Off | 00000000:1A:00.0 Off | N/A | | 73% 70C P0 73W / 250W | 0MiB / 10989MiB | 0% Default | +-------------------------------+----------------------+----------------------+ | 2 GeForce RTX 208... Off | 00000000:67:00.0 Off | N/A | | 79% 71C P0 86W / 250W | 0MiB / 10989MiB | 1% Default | +-------------------------------+----------------------+----------------------+ | 3 GeForce RTX 208... Off | 00000000:68:00.0 Off | N/A | | 44% 71C P0 1W / 250W | 0MiB / 10986MiB | 0% Default | +-------------------------------+----------------------+----------------------+ +-----------------------------------------------------------------------------+ | Processes: GPU Memory | | GPU PID Type Process name Usage | |=============================================================================| | No running processes found | +-----------------------------------------------------------------------------+