[Ubuntu 16.04~] CUDA 10.2, cuDNN 7.6.5 설치

CPU : i8-8700K / GPU : GeForce GTX1080 Ti 기준

default로 설치한 ubuntu 16.04에 CUDA를 그대로 설치하면 무한 로그인같은 이상한 현상이 발생하는데

ubuntu 설치할때 커널 부팅옵션을 추가해줘야한다.

EFI 모드로 설치 시

Install Ubuntu에 포커스를 맞추고 e키를 누르면 아래와 같은 창이 뜨는데

linux 라인 끝(quiet splash와 ---사이)에 다음과 같이 명령어를 추가해주고 ctrl+x 또는 F10 키를 입력하여 설치를 진행한다.

nomodeset acpi=off

Legacy 설치 시

키보드의 F6키를 누르면 메뉴 팝업이 뜨고 마찬가지로 nomodeset, acpi=off 항목을 체크 (화살표 키로 이동 후 Enter)한 뒤 Install Ubuntu를 선택하여 설치를 진행한다.

CUDA 10.2 설치

CUDA Toolkit Download 사이트에 들어가 본인이 사용중인 운영체제와 아키텍쳐, 버전을 선택한다. deb(local)을 클릭했을 때 뜨는 명령어를 그대로 입력해준다.

wget https://developer.download.nvidia.com/compute/cuda/repos/ubuntu1604/x86_64/cuda-ubuntu1604.pin
sudo mv cuda-ubuntu1604.pin /etc/apt/preferences.d/cuda-repository-pin-600
wget http://developer.download.nvidia.com/compute/cuda/10.2/Prod/local_installers/cuda-repo-ubuntu1604-10-2-local-10.2.89-440.33.01_1.0-1_amd64.deb
sudo dpkg -i cuda-repo-ubuntu1604-10-2-local-10.2.89-440.33.01_1.0-1_amd64.deb
sudo apt-key add /var/cuda-repo-10-2-local-10.2.89-440.33.01/7fa2af80.pub
sudo apt-get update
sudo apt-get -y install cuda

파일 하나 다운 받고 설치가 종료되는데 설치를 확인하기 위해 아래 명령어를 입력

nvidia-smi

대충 이런 메시지가 뜨면서 드라이버 버전 등 정보가 뜨면 일단 CUDA는 설치 성공

cuDNN 7.6.5 설치

cuDNN은 Nvidia Developer Program가입이 필요하다.
https://developer.nvidia.com/rdp/form/cudnn-download-survey 여기 가서 Join now를 클릭해 가입 혹은 Login 하자

로그인 했다면 약관에 동의를 해주고 CUDA 버전에 맞는 cuDNN을 찾아 클릭하고

cuDNN Library for Linux를 선택하여 압축파일을 받는다.

바로 위 Installation Guide 참고해서 설치하면 되지만 귀찮으면 다운로드 받은 폴더로 가서 아래 명령어를 그대로 복붙하면 된다.

tar -xzvf cudnn-10.2-linux-x64-v7.6.5.32.tgz
sudo cp cuda/include/cudnn.h /usr/local/cuda/include
sudo cp cuda/lib64/libcudnn* /usr/local/cuda/lib64
sudo chmod a+r /usr/local/cuda/include/cudnn.h /usr/local/cuda/lib64/libcudnn*

그 다음 아래 두 줄을 ~/.bashrc에 추가해주자.

export PATH=/usr/local/cuda-10.2/bin${PATH:+:${PATH}}
export LD_LIBRARY_PATH=/usr/local/cuda-10.2/lib64\${LD_LIBRARY_PATH:+:${LD_LIBRARY_PATH}}

libcudnn.so.7 올바른 심볼릭 링크가 아니라는 경고가 뜰 때는 그냥 아래 명령어로 libcudnn.so.7.6.5 링크를 다시 걸어주면 된다고 한다.

sudo ln -sf /usr/local/cuda-10.2/targets/x86_64-linux/lib/libcudnn.so.7.6.5 /usr/local/cuda-10.2/targets/x86_64-linux/lib/libcudnn.so.7

이후 아래 명령어를 입력하면 CUDA와 cuDNN 버전 정보가 뜨면서 설치가 완료된다.

nvcc --version

nvcc: NVIDIA (R) Cuda compiler driver
Copyright (c) 2005-2019 NVIDIA Corporation
Built on Wed_Oct_23_19:24:38_PDT_2019
Cuda compilation tools, release 10.2, V10.2.89