Ubuntu 18.04 | CUDA & cuDNN 설치

Ubuntu 18.04에 CUDA와 cuDNN 설치하기

CUDA 버전은 11.0

cuDNN 버전은 v8.0.5

 

 

Linux -> x86_64 -> Ubuntu -> 18.04 -> deb (local)

 

CUDA를 다운로드 하기 위해 여기로 이동.

 

CUDA deb(local) 설치

$ wget https://developer.download.nvidia.com/compute/cuda/repos/ubuntu1804/x86_64/cuda-ubuntu1804.pin
$ sudo mv cuda-ubuntu1804.pin /etc/apt/preferences.d/cuda-repository-pin-600
$ wget http://developer.download.nvidia.com/compute/cuda/11.0.2/local_installers/cuda-repo-ubuntu1804-11-0-local_11.0.2-450.51.05-1_amd64.deb
$ sudo dpkg -i cuda-repo-ubuntu1804-11-0-local_11.0.2-450.51.05-1_amd64.deb
$ sudo apt-key add /var/cuda-repo-ubuntu1804-11-0-local/7fa2af80.pub
$ sudo apt-get update
$ sudo apt-get -y install cuda

 

환경변수 설정

$ gedit ~/.profile # 환경변수 설정

# 아래 내용 입력 후 저장
export PATH=/usr/local/cuda-10.1/bin:$PATH
export LD_LIBRARY_PATH=/usr/local/cuda-10.1/lib64:$LD_LIBRARY_PATH

$ source ~/.profile # 환경변수 설정 적용

$ sudo reboot # 재부팅

 

CUDA 설치 확인

$ nvcc --version
$ nvidia-smi

 

 

cuDNN을 설치하기 위해 여기로 이동.

 

cuDNN 설치

$ cd ~/Downloads
$ tar -zxvf cudnn-11.0-linux-x64-v8.0.5.39.tgz
$ sudo cp ./cuda/include/* /usr/local/cuda-11.0/include # cuDNN header 파일들 복사
$ sudo cp -P ./cuda/lib64/* /usr/local/cuda-11.0/lib64 # cuDNN  library 파일들 복사
$ sudo chmod a+r /usr/local/cuda-11.0/lib64/libcudnn*

 

cuDNN 설치 확인

$ cat /usr/local/cuda/include/cudnn.h | grep CUDNN_MAJOR -A 2 (cuDNN 7.x 까지)
$ cat /usr/local/cuda/include/cudnn_version.h | grep CUDNN_MAJOR -A 2 (cuDNN 8.x 부터)

 

cuDNN 설치 확인 시 아래와 같이 출력이 됨

#define CUDNN_MAJOR 8
#define CUDNN_MINOR 0
#define CUDNN_PATCHLEVEL 5
--
#define CUDNN_VERSION (CUDNN_MAJOR * 1000 + CUDNN_MINOR * 100 + CUDNN_PATCHLEVEL)

#endif /* CUDNN_VERSION_H */