Ubuntu 환경에서 NVIDIA 드라이버 설치

Ubuntu에서 NVIDIA Driver를 설치하기

이글은 인공지능(머신러닝)을 위해서 GPU 환경을 꾸미는 분이 가장 많이 사용하는 NVIDIA 장치 드라이버를 설치하는 과정을 정리한 것입니다.

  • 사용환경
  • ubuntu 14.04.05

Download NVIDIA 드라이버

  • 2017년 01월 31일 update: NVIDIA-Linux-x86_64-375.26.run
  • 2016년 09월 시점 참조: NVIDIA-Linux-x86_64-367.44.run
  • 필요한 패키지 설치

sudo apt-get install dkms build-essential linux-headers-generic

  • modprobe에서 설정 파일 추가

sudo vi /etc/modprobe.d/blacklist-nouveau.conf

blacklist nouveau
blacklist lbm-nouveau
options nouveau modeset=0
alias nouveau off
alias lvm-nouveau off
  • nouveau 설정 제거

echo options nouveau modeset=0 | sudo tee -a /etc/modprobe.d/nouveau-kms.conf

  • nouveau 설정 업데이트

sudo update-initramfs -u

  • 다운 받은 NVIDIA 드라이버를 실행 파일로 바꾸고, 실행

chmod +x NVIDIA-Linux--375.26.run && sudo sh NVIDIA-Linux--375.26.run

선택과정에서 다음처럼 선택

Select Accept Select Yes to Kernel DKMS Select Ok if prompts that no 32-bit compatibility. Select Yes to configuration X

  • 마무리 작업

cd /etc/modprobe.d/ && sudo rm blacklist-nouveau.conf nouveau-kms.conf && sudo update-initramfs -u

드라이버 설치 확인

$ nvidia-smi

  • 2016년 1월 31일 (update)
Tue Jan 31 15:33:51 2017       
+-----------------------------------------------------------------------------+
| NVIDIA-SMI 375.26                 Driver Version: 375.26                    |
|-------------------------------+----------------------+----------------------+
| 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 TIT...  Off  | 0000:01:00.0     Off |                  N/A |
|  0%   58C    P0    57W / 250W |      0MiB / 12198MiB |      0%      Default |
+-------------------------------+----------------------+----------------------+
                                                                               
+-----------------------------------------------------------------------------+
| Processes:                                                       GPU Memory |
|  GPU       PID  Type  Process name                               Usage      |
|=============================================================================|
|  No running processes found                                                 |
+-----------------------------------------------------------------------------+

다음 과정

인공지능(머신러닝)을 위한 환경 설정을 위해서는 CUDA, CuDNN 설정이 추가로 필요합니다.

2016년 9월 시점 (구버전)

Tue Sep 20 22:01:45 2016       
+-----------------------------------------------------------------------------+
| NVIDIA-SMI 367.44                 Driver Version: 367.44                    |
|-------------------------------+----------------------+----------------------+
| 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 970     Off  | 0000:01:00.0     Off |                  N/A |
|  0%   43C    P0    38W / 151W |      0MiB /  4028MiB |      0%      Default |
+-------------------------------+----------------------+----------------------+
                                                                               
+-----------------------------------------------------------------------------+
| Processes:                                                       GPU Memory |
|  GPU       PID  Type  Process name                               Usage      |
|=============================================================================|
|  No running processes found                                                 |
+-----------------------------------------------------------------------------+