Ubuntu CUDA 설치하기

Ubuntu 환경에서 딥러닝을 처음 시작하는 사람들을 위한 포스팅이다.

 

1. 먼저 현재 컴퓨터에 설치된 그래픽 드라이버 버전을 확인해야 한다.

 

터미널 실행 후 아래 명령어 실행

$ nvidia-smi

그래픽 드라이버가 설치되어 있지 않으면 위 그림 상태가 뜨지 않는다.

 

그래픽 드라이버 설치 홈페이지

https://www.nvidia.co.kr/Download/index.aspx?lang=kr

 

특히 그래픽 드라이버 설치 중 에러가 많다.

 

X Server Error

해결 방법:

1) Ctrl + Alt + F1                    # 터미널 창으로 이동 (F1이 안될때 다른 F2,3,4,5 키 눌러보기)

2) $ sudo service lightdm stop   # 그래픽 화면 끄기

3) $ cd ./Downloads                # 다운로드한 드라이브 설치 파일이 있는곳으로 이동 (보통 Downloads에 저장됨)

4) $ sudo chmod +x ./(NVIDIA_FILE)   # 드라이브 설치 파일을 실행할 수 있게 권한 변경

5) $ sudo ./(NVIDIA_FILE)          # 드라이브 설치 파일 실행

6) $ reboot                            #  재부팅하기

 

Nouveau Error

해결 방법:

1) $ cd /etc/modprobe.d/                          # 파일로 이동

2) $ sudo gedit blacklist-nouveau.conf          # 파일 생성

3) 밑에 내용 입력 및 저장
    blacklist nouveau
    options nouveau modeset=0

4) $ sudo update-initramfs -u                     # 변경된 내용 업로드

5) $ reboot                                             # 재부팅하기

 

2. CUDA를 설치하기 이전 그래픽 드라이버 버전에서 지원하는 CUDA Version을 확인하자.

Graphic Driver Version에 따른 CUDA 버전 호환 

 

이 포스팅에서는 CUDA 10.0 설치해본다. 

$ sudo apt-get install cuda-10-0

~/.bashrc 경로를 추가해야한다. 

1) $ sudo gedit ~/.bashrc

2) 아래 내용을 추가해준다.
export PATH=/usr/local/cuda-10.0/bin${PATH:+:${PATH}}
export LD_LIBRARY_PATH=/usr/local/cuda/lib64:${LD_LIBRARY_PATH:+:${LD_LIBRARY_PATH}}
# 주의) 첫줄의 cuda 버전에 맞추어서 입력해야됨

 

다음은 cuDNN을 설치합니다.

$ sudo apt-get install libcudnn7-dev 

※ cudnn 수동 설치 방법

1) https://developer.nvidia.com/rdp/form/cudnn-download-survey     # Login후 CUDA 버전에 맞는 cudnn 설치

2) $ cd ./Downloads            # 설치 파일 경로로 이동

3) $ sudo tar -xzvf (your cudnn).tgz

4) $ cd cuda

5) $ sudo cp include/cudnn.h /usr/local/cuda/include

6) $ sudo cp lib64/libcudnn* /usr/local/cuda/lib64

7) $ sudo chmod a+r /usr/local/cuda/lib64/libcudnn*

8) $ cat /usr/local/cuda/include/cudnn.h | grep CUDNN_MAJOR -A 2