IT/YOLO

[2] Ubuntu 16.04(64bit), CUDA 8.0, cuDNN 5.1 설치

주니- 2018. 6. 16. 01:00

우분투에서 패키지 설치 전에는 항상 패키지를 최신으로 만들어주셔야합니다.

 

해당 명령어를 통해 항상 최신으로 업데이트 해주세요. 

1
2
sudo apt-get update
sudo apt-get upgrade

NVIDIA 그래픽 드라이버 설치를 위하여 다음 명령어를 입력해주세요.

1
2
3
sudo apt-get install software-properties-common
sudo add-apt-repository ppa:graphics-drivers/ppa
sudo apt-get update

완료되면 왼쪽 탭 메뉴에 톱니바퀴 모양의 아이콘으로 되어있는 System Settings(시스템 설정)에 들어가서 Software & Updates(소프트웨어 & 업데이트) -> Additional Drivers(추가 드라이버) 탭으로 이동해주세요.
CUDA 8.0은 최소 367.4x 이상의 드라이버가 필요하다고 합니다. 저는 384로 설치했습니다.
같이 선택한 후 Apply Changes(바뀐 내용 적용) 버튼을 클릭해 설치해주세요. 설치가 끝나면 reboot(재시작)을 해주세요.

재부팅이 완료되고 터미널에서

1
nvidia-settings

를 입력하면 NVIDIA X Server Settings가 실행됩니다. 이러면 드라이버 설치가 완료되었습니다.

이제 CUDA ToolKit을 설치하시면 됩니다. 제가 설치한 버전은 8.0입니다. (저도 블로그 참고 하였습니다.)

CUDA Download에서 회원가입을 하고 로그인을 한 뒤

Select Target Platform 메뉴에서 Linux -> x86_64 -> Ubuntu -> 16.04 -> runfile(local)을 다운받으세요.

다운이 완료되고 경로로가서 runfile을 실행합니다.

1
2
cd <다운로드 경로>
sudo sh cuda_8.0.44_linux.run

실행하면 End User License Agreement가 나오게 되는데 이때 Ctrl+c를 누르면 다음과 같은 질문이 나오고

1
2
3
4
5
6
7
8
9
10
11
12
13
14
accept/decline/quit: accept
이후엔 옵션을 설정하는 부분으로
Install NVIDIA Accelerated Graphics Driver for Linu….?이 질문은 이미 위에서 설치했기 때문에
(y)es/(n)o/(q)uit: n
Install the CUDA 8.0 Toolkit? 당연히 대답은
(y)es/(n)o/(q)uit: y
Enter Toolkit Location [ default is /usr/local/cuda-8.0 ]:
엔터를 쳐서 default를 사용한다.
Do you want to install a symbolic link at /usr/local/cuda?
(y)es/(n)o/(q)uit: y
Install the CUDA 8.0 Samples?
(y)es/(n)o/(q)uit: y
Enter CUDA Samples Location [ default is /home/사용자 이름 ]:
엔터를 쳐서 default를 사용한다.

설치가 진행되고 끝나면 Summary를 통해 결과를 확인할 수 있습니다. 우리가 확인해야할 Summary는

1
2
3
Driver: Not Selected
Toolkit: Installed in /usr/local/cuda-8.0
Samples: Installed in /home/사용자 이름

이런 내용이 나오면 설치가 정상적으로 완료된 것입니다.
이제 PATH variable을 설정해주세요.

1
2
3
4
export PATH=/usr/local/cuda-8.0/bin${PATH:+:${PATH}}
export LD_LIBRARY_PATH=/usr/local/cuda-8.0/lib64\${LD_LIBRARY_PATH:+:${LD_LIBRARY_PATH}}
source ~/.bashrc
nvcc --version

이렇게 출력되면 됩니다.

1
2
3
4
nvcc: NVIDIA (R) Cuda compiler driver
Copyright (c) 2005-2016 NVIDIA Corporation
Built on Sun_Sep__4_22:14:01_CDT_2016
Cuda compilation tools, release 8.0, V8.0.44

마지막으로 cuDNN 5.1을 설치해주세요.
cuDNN Download 에서 cuDNN v5.1CUDA 8.0에 해당하는 버튼을 누르고
cuDNN v5.1 Library for Linux를 클릭해 다운받아주세요.  

1
2
3
4
5
6
7
cd <다운로드 경로>
sudo tar -xzvf cudnn-8.0-linux-x64-v5.1.tgz
cd cuda
sudo cp include/cudnn.h /usr/local/cuda/include
sudo cp lib64/libcudnn* /usr/local/cuda/lib64
sudo chmod a+r /usr/local/cuda/lib64/libcudnn*
cat /usr/local/cuda/include/cudnn.h | grep CUDNN_MAJOR -A 2

 이때 출력 화면에 CUDNN_MAJOR가 포함되어 있으면 제대로 설치된 것입니다.

 

 

※ 도와주신 rookieyun님 감사합니다.