Ubuntu Linux 18.04にGPUドライバをインストールする

上記でLinuxをインストールしましたが、GPUのドライバがインストールされていないため、ディスプレイに表示される画面も解像度が低く、またこのままでは機械学習などの仕事をするときにGPUが使えないため、NVIDIAのGPU RTX 2070のドライバーをインストールします。

はじめに、aptのリポジトリにNVIDIAのドライバのリポジトリを追加。ただし、ppaなので自己責任の元で追加です。以下を確認してください。

Proprietary GPU Drivers : “Graphics Drivers” team

sudo add-apt-repository ppa:graphics-drivers/ppa

その後、aptをアップデート

sudo apt update

つぎに、推奨ドライバーを確認

ubuntu-drivers devices

以下のような結果がでて推奨ドライバは、430の方でしたが、

masato1@masato1-ubuntu:~$ ubuntu-drivers devices
== /sys/devices/pci0000:00/0000:00:01.0/0000:01:00.0 ==
modalias : pci:v000010DEd00001F02sv000019DAsd00002516bc03sc00i00
vendor   : NVIDIA Corporation
driver   : nvidia-driver-418 - third-party free
driver   : nvidia-driver-410 - third-party free
driver   : nvidia-driver-430 - third-party free recommended
driver   : nvidia-driver-415 - third-party free
driver   : xserver-xorg-video-nouveau - distro free builtin

masato1@masato1-ubuntu:~$ 

NVIDIA DRIVERS Linux x64 (AMD64/EM64T) Display Driver

で検索したときは418の方が推奨されていて、なんとなく最新版だとハマりそうなので、バグでも踏むまでは418の方を使おうと思うことにして、418の方をインストールしました。

sudo apt install nvidia-driver-418

再起動するとnvidia-smiコマンドが使えて、ドライバが機能しているのがわかります。

masato1@masato1-ubuntu:~$ nvidia-smi
Thu May  9 18:07:50 2019       
+-----------------------------------------------------------------------------+
| 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 2070    Off  | 00000000:01:00.0  On |                  N/A |
| 40%   50C    P0    46W / 175W |    237MiB /  7944MiB |      0%      Default |
+-------------------------------+----------------------+----------------------+
                                                                               
+-----------------------------------------------------------------------------+
| Processes:                                                       GPU Memory |
|  GPU       PID   Type   Process name                             Usage      |
|=============================================================================|
|    0      1121      G   /usr/lib/xorg/Xorg                            18MiB |
|    0      1158      G   /usr/bin/gnome-shell                          50MiB |
|    0      1442      G   /usr/lib/xorg/Xorg                            96MiB |
|    0      1557      G   /usr/bin/gnome-shell                          70MiB |
+-----------------------------------------------------------------------------+
masato1@masato1-ubuntu:~$ 

以下のマガジンで細々と機械学習環境を作る過程をアップデートしてます。

自宅で機械学習環境構築｜setoyama60jp｜note