머신러닝 준비 1/4: Ubuntu에 NVIDIA 드라이버 설치

머신러닝을 위한 대부분의 라이브러리들은 Linux를 기반으로 하고 있다.

VMWare상으로는 GPU 가속을 받을 수 없기 때문에 별도의 머신이 필요하다.

물론 Docker를 기반으로 1대의 윈도우 PC로 구동 가능하나 100% 성능을 뽑아낼 수 없다고 본 바

별도의 머신을 셋팅하는 것을 추천한다.

리눅스는 다양한 OS들이 존재하는데 보편화된 것들(CentOS, Ubuntu) 중 Ubuntu로 머신을 셋팅 하도록 결정했다.

리눅스 머신을 셋팅하는 방법은 다음과 같다.

1. Ubuntu 64bit 설치

우분투를 다운로드 하여 설치한다.

그래픽관련 프로그램들을 설치해야 하므로 Ubuntu Server가 아닌 Ubuntu Desktop으로 설치 하도록 한다.

이 글을 쓰면서 설치 한 버전은 Ubuntu 16.04.1 LTS 이다.

주의! 최신 그래픽카드를 장착한 상태에서 설치 시 모니터가 인식하지 못하는 문제가 발생 할 수 있다.

그럴 땐 일단 최신 그래픽 카드를 탈착 후 온보드 그래픽카드로 Ubuntu를 설치한다.

그리고 후에 그래픽카드를 장착 하고 부팅하면 기본 640 * 480 해상도로 정상 부팅이 가능하다.

2. NVIDIA 드라이버 설치

NVIDIA 드라이버는 직접 다운로드 하는 방법과 자동 설치 방법으로 나뉜다. 원하는 스타일 대로 실행한다.

1. 드라이버 자동 설치

아래 명령어를 통해 한번에 설치 가능하다. (2번 과정을 건너뛰어도 된다.)

$ sudo apt-get install nvidia-current nvidia-settings

2. 드라이버 다운로드 설치

NVIDIA 드라이버 다운로드 사이트에서 최신 드라이버를 다운로드 한다.

다운로드된 드라이버는 알맞은 영문명 폴더(/tmp 등)에 이동 해 놓는다.

2-1. tty1 접속

Ubuntu Desktop에서 Ctrl + Alt + F1키를 눌러 tty1 콘솔로 접속한다.

콘솔화면이 뜨게 되는데 로그인 후 드라이버가 있는 위치로 이동한다.

$ cd /tmp

2-2. 드라이버 설치

$ sudo service lightdm stop

$ sudo sh NVIDIA_드라이버_파일이름.run

$ sudo service lightdm start

3. 드라이버 설치 확인

콘솔창을 띄워 다음과 같이 명령어를 실행한다.

NVIDIA 드라이버 이름이 뜬다면 성공적으로 설치가 된 것이다.

$ cat /proc/driver/nvidia/version