Ubuntu 18.04 cuDNN 설치

cuDNN

CUDA기반의 Deep Neural Network 라이브러리를 의미한다.

cuDNN을 설치하기전 Nvidia driver와 CUDA를 설치해야 한다.

---

로그인

cuDNN을 다운받기 위해서는 Nvida에 회원 가입을 하고 로그인 해야 한다. (필수)

cuDNN 8.0.2 라이브러리 다운로드 

https://developer.nvidia.com/rdp/cudnn-archive

cuDNN Archive

해당 주소로 접속해 CUDA버전에 맞는 cuDNN 라이브러리를 설치해야 한다.

해당 블로그 에서는 Download cuDNN v8.0.2 (July 24th, 2020), for CUDA 10.2 선택하고 cuDNN Library for Linux (x86) 다운로드하겠다.

cd 다운로드

tar -xvf cudnn-10.2-linux-x64-v8.0.2.39.tgz

sudo cp ./cuda/include/* /usr/local/cuda-10.2/include

sudo cp -P ./cuda/lib64/* /usr/local/cuda-10.2/lib64

sudo chmod a+r /usr/local/cuda-10.2/lib64/libcudnn*

---

cuDNN install check!

cuDNN 8.x 이전 버전에서 cat /usr/local/cuda/include/cudnn.h 에서 버전 확인

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

cuDNN 8.x 이상 버전에서는 cudnn_version.h 파일에서 버전 확인

 $ cat /usr/local/cuda/include/cudnn_version.h | grep CUDNN_MAJOR -A 2

 

버전 확인 결과

#define CUDNN_MAJOR 8
#define CUDNN_MINOR 2
#define CUDNN_PATCHLEVEL 0
--
#define CUDNN_VERSION (CUDNN_MAJOR * 1000 + CUDNN_MINOR * 100 + CUDNN_PATCHLEVEL)
#endif /* CUDNN_VERSION_H */