[Ubuntu 18.04] Nvidia GPU 드라이버 설치

Ubuntu를 설치하고 머신러닝 등의 학습을 진행하기 위해서는 먼저 GPU Driver가 설치되어야 합니다.

 

장치확인

먼저 "lshw -C display" 명령으로 GPU 장치가 잘 인식되어있는지 확인합니다.

# lshw -C display
WARNING: you should run this program as super-user.
  *-display                 
       description: VGA compatible controller
       product: GP106 [GeForce GTX 1060 3GB]
       vendor: NVIDIA Corporation
       physical id: 0
       bus info: pci@0000:0b:00.0
       version: a1
       width: 64 bits
       clock: 33MHz
       capabilities: vga_controller bus_master cap_list rom
       configuration: driver=nvidia latency=0

제 PC에는 GTX 1060이 설치되어있습니다.

간혹, 미표시 또는 내장그래픽이 인식되는 경우가 있는데 드라이버를 일단 설치해보면 괜찮아질 수 있습니다.

권장설치

Ubuntu에서는 드라이버 설치를 위하여 "ubuntu-drivers"라는 명령을 제공합니다.

설치가능 목록은 "ubuntu-drivers devices"로, 권장설치는 "ubuntu-drivers autoinstall"로 진행합니다.

# sudo ubuntu-drivers devices
== /sys/devices/pci0000:00/0000:00:03.1/0000:0b:00.0 ==
modalias : pci:v000010DEd00001C02sv00001462sd00003287bc03sc00i00
vendor   : NVIDIA Corporation
model    : GP106 [GeForce GTX 1060 3GB]
driver   : nvidia-driver-430 - distro non-free
driver   : nvidia-driver-440 - third-party free recommended
driver   : nvidia-driver-410 - third-party free
driver   : nvidia-driver-435 - distro non-free
driver   : nvidia-driver-390 - distro non-free
driver   : xserver-xorg-video-nouveau - distro free builtin

# sudo ubuntu-drivers autoinstall

선택설치

특정버전 설치가 필요한 경우에는 "apt-get install nvidia-driver-[버전]"을 활용하시면 됩니다.

"sudo apt-cache search NVIDIA driver metapackage"를 활용하여 설치가능한 버전을 확인할 수 있습니다.

예전과는 다르게 ppa를 별도로 추가하실 필요는 없습니다.

# sudo apt-cache search NVIDIA driver metapackage

vdpau-driver-all - Video Decode and Presentation API for Unix (driver metapackage)
nvidia-driver-390 - NVIDIA driver metapackage
nvidia-headless-390 - NVIDIA headless metapackage
nvidia-headless-no-dkms-390 - NVIDIA headless metapackage - no DKMS
bumblebee-nvidia - NVIDIA Optimus support using the proprietary NVIDIA driver
nvidia-driver-430 - NVIDIA driver metapackage
nvidia-driver-435 - NVIDIA driver metapackage
nvidia-headless-430 - NVIDIA headless metapackage
nvidia-headless-435 - NVIDIA headless metapackage
nvidia-headless-no-dkms-430 - NVIDIA headless metapackage - no DKMS
nvidia-headless-no-dkms-435 - NVIDIA headless metapackage - no DKMS
nvidia-driver-410 - NVIDIA driver metapackage
nvidia-driver-440 - NVIDIA driver metapackage
nvidia-headless-410 - NVIDIA headless metapackage
nvidia-headless-440 - NVIDIA headless metapackage
nvidia-headless-no-dkms-410 - NVIDIA headless metapackage - no DKMS
nvidia-headless-no-dkms-440 - NVIDIA headless metapackage - no DKMS

# sudo apt-get install nvidia-driver-440
....

설치확인

드라이버를 설치하고 재부팅을 완료하면 정상적으로 설치가 되었을 것 입니다.

"nvidia-smi" 명령을 통해 정상적으로 인식하고 있는지 확인하시면 됩니다.

# nvidia-smi
Tue Feb 11 01:16:32 2020       
+-----------------------------------------------------------------------------+
| NVIDIA-SMI 440.33.01    Driver Version: 440.33.01    CUDA Version: 10.2     |
|-------------------------------+----------------------+----------------------+
| 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 GTX 106...  On   | 00000000:0B:00.0  On |                  N/A |
| 35%   35C    P8     6W / 120W |    566MiB /  3016MiB |      1%      Default |
+-------------------------------+----------------------+----------------------+
                                                                               
+-----------------------------------------------------------------------------+
| Processes:                                                       GPU Memory |
|  GPU       PID   Type   Process name                             Usage      |
|=============================================================================|
|    0      1239      G   /usr/lib/xorg/Xorg                            26MiB |
|    0      1284      G   /usr/bin/gnome-shell                          49MiB |
|    0      1835      G   /usr/lib/xorg/Xorg                           229MiB |
|    0      1980      G   /usr/bin/gnome-shell                         149MiB |
|    0      2560      G   ...AAAAAAAAAAAAAAgAAAAAAAAA --shared-files   105MiB |
+-----------------------------------------------------------------------------+