1. 查看ubuntu系统使用的显卡型号
lspci | grep VGA
2.检查驱动程序是否安装
nvidia-smi
如果没有显示以下类型的界面,说明显卡驱动未安装
hkc@NODE-021:~$ nvidia-smi
Fri Mar 5 08:56:07 2021
+-----------------------------------------------------------------------------+
| NVIDIA-SMI 460.39 Driver Version: 460.39 CUDA Version: 11.2 |
|-------------------------------+----------------------+----------------------+
| GPU Name Persistence-M| Bus-Id Disp.A | Volatile Uncorr. ECC |
| Fan Temp Perf Pwr:Usage/Cap| Memory-Usage | GPU-Util Compute M. |
| | | MIG M. |
|===============================+======================+======================|
| 0 GeForce RTX 208... Off | 00000000:01:00.0 Off | N/A |
| 41% 33C P8 20W / 257W | 15MiB / 11018MiB | 0% Default |
| | | N/A |
+-------------------------------+----------------------+----------------------+
+-----------------------------------------------------------------------------+
| Processes: |
| GPU GI CI PID Type Process name GPU Memory |
| ID ID Usage |
|=============================================================================|
| 0 N/A N/A 1033 G /usr/lib/xorg/Xorg 9MiB |
| 0 N/A N/A 1399 G /usr/bin/gnome-shell 3MiB |
+-----------------------------------------------------------------------------+
3.检查适合的驱动程序版本
ubuntu-drivers devices
4.安装驱动程序
两种方法:
A.sudo apt install nvidia-driver-xxx(xxx为第3步系统推荐的驱动程序版本)
B.sudo ubuntu-drivers autoinstall(自动安装适合的驱动程序)
安装完成后需要重启。
重启完后执行nvidia-smi就可以显示显卡情况了
hkc@NODE-021:~$ nvidia-smi
Fri Mar 5 08:56:07 2021
+-----------------------------------------------------------------------------+
| NVIDIA-SMI 460.39 Driver Version: 460.39 CUDA Version: 11.2 |
|-------------------------------+----------------------+----------------------+
| GPU Name Persistence-M| Bus-Id Disp.A | Volatile Uncorr. ECC |
| Fan Temp Perf Pwr:Usage/Cap| Memory-Usage | GPU-Util Compute M. |
| | | MIG M. |
|===============================+======================+======================|
| 0 GeForce RTX 208... Off | 00000000:01:00.0 Off | N/A |
| 41% 33C P8 20W / 257W | 15MiB / 11018MiB | 0% Default |
| | | N/A |
+-------------------------------+----------------------+----------------------+
+-----------------------------------------------------------------------------+
| Processes: |
| GPU GI CI PID Type Process name GPU Memory |
| ID ID Usage |
|=============================================================================|
| 0 N/A N/A 1033 G /usr/lib/xorg/Xorg 9MiB |
| 0 N/A N/A 1399 G /usr/bin/gnome-shell 3MiB |
+-----------------------------------------------------------------------------+
5. 卸载显卡驱动
sudo apt-get remove nvidia* -y