ubuntu20.04にNVIDIA GPUドライバをインストール

• 動作環境
  • OS: Ubuntu 20.04.2 LTS (focal)
  • GPU: GeForce GTX 960

忙しい人のために

結果から言うと、以下のコマンドを叩けばインストールされる

sudo apt update -y
sudo apt install -y ubuntu-drivers-common
driver=$(ubuntu-drivers devices | grep recommended | awk '{print $3}')
sudo apt install -y --no-install-recommends $driver

インストール方法

今回はaptでインストールする方法について書く。
※ ソースビルドやアンインストールの方法については書かない。

まずは、推奨ドライバを調べる。
調べるためのubuntu-driversコマンドを入れて、コマンドを叩く。

$ sudo apt install -y ubuntu-drivers-common
$ ubuntu-drivers devices
== /sys/devices/pci0000:00/0000:00:03.1/0000:0e:00.0 ==
modalias : pci:xxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxx
vendor   : NVIDIA Corporation
model    : GM206 [GeForce GTX 960]
driver   : nvidia-driver-460-server - distro non-free
driver   : nvidia-driver-418-server - distro non-free
driver   : nvidia-driver-460 - distro non-free recommended
driver   : nvidia-driver-390 - distro non-free
driver   : nvidia-driver-450-server - distro non-free
driver   : nvidia-driver-450 - distro non-free
driver   : xserver-xorg-video-nouveau - distro free builtin

この中から"recommended"が付いているものをインストールする。
--no-install-recommendsをつけて、余計なパッケージ（デスクトップ・オフィス等）が入らないようにする。

sudo apt install -y --no-install-recommends nvidia-driver-460

OSを再起動し、nvidia-smiコマンドでインストールされていることを確認。

$ nvidia-smi 
Mon Mar 15 17:41:20 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 GTX 960     Off  | 00000000:0E:00.0 Off |                  N/A |
| 30%   54C    P2    38W / 130W |    652MiB /  4042MiB |     47%      Default |
|                               |                      |                  N/A |
+-------------------------------+----------------------+----------------------+

+-----------------------------------------------------------------------------+
| Processes:                                                                  |
|  GPU   GI   CI        PID   Type   Process name                  GPU Memory |
|        ID   ID                                                   Usage      |
|=============================================================================|
|    0   N/A  N/A   1998526      C   /usr/bin/ffmpeg                   215MiB |
|    0   N/A  N/A   2011374      C   /usr/bin/ffmpeg                   215MiB |
|    0   N/A  N/A   2011908      C   /usr/bin/ffmpeg                   215MiB |
+-----------------------------------------------------------------------------+