Ubuntu에 cuDNN 설치

Ubuntu에 cuDNN 설치

 

이전 포스팅에서 CUDA를 설치해 보았습니다.

Ubuntu에 CUDA 설치

 

 

이번 포스팅에서는 GPU 가속 라이브러리인 cuDNN을 설치해 보겠습니다.

 

아래 사이트에서 cuDNN가 CUDA의 버전 호환성을 확인할 수 있습니다.

Support Matrix — NVIDIA cuDNN

 

 

버전 호환성을 확인하고 아래 사이트에서 cuDNN을 설치할 수 있습니다.

cuDNN Archive

 

 

저는 CUDA 12.4가 설치되어 있기 때문에 cuDNN 9.5.1 버전을 설치해 보겠습니다.

설치할 버전을 클릭하고, CUDA를 설치했을 때와 같이 본인 PC 환경에 맞게 설정합니다.

저는 Linux > x86_64 > Ubuntu > 20.04 > deb (local) 순으로 설정하였습니다.

cuDNN Download

 

 

설정하면 아래에 명령어들이 나타나는데 순서대로 터미널에 입력하면 cuDNN이 설치됩니다.

만약 CUDA 11.X 버전을 사용하고 있으면 cudnn-cuda-11을 다운 받아야 합니다.

cuDNN Download

 

 

설치가 끝나면 다음 명령어를 통해 버전을 확인할 수 있습니다.

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

cuDNN 버전 확인

 

---

cuDNN 8.x - 1.x 버전 설치하는 방법

cuDNN 8.x - 1.x 버전은 아래 사이트에서 설치할 수 있습니다.

cuDNN Archive

 

 

여기서 원하는 버전을 고르고 Linux x86_64 (Tar) 파일을 다운 받습니다.

cuDNN Download

 

 

설치한 파일 경로인 Download 폴더에 들어가서 다운 받은 Tar 파일의 압축을 풉니다.

그리고 터미널을 열고 아래 명령어들을 실행합니다.본인이 설치한 버전에 맞게 괄호 안을 수정하면 됩니다.

cd Downloads/<압축 해제한 폴더명>/
sudo cp include/cudnn* /usr/local/cuda-<버전>/include
sudo cp lib/libcudnn* /usr/local/cuda-<버전>/lib64/
sudo chmod a+r /usr/local/cuda-<버전>/lib64/libcudnn*

 

 

설치가 되었는지 확인하기 위해 아래 명령어를 실행합니다.

이것도 본인이 설치한 버전에 맞게 괄호 안을 수정하면 됩니다.

cat /usr/local/cuda-<버전>/include/cudnn_version.h | grep CUDNN_MAJOR -A 2

 

 

실행 결과, 다음과 같이 설치한 cuDNN 버전을 확인할 수 있습니다.

#define CUDNN_MAJOR 9
#define CUDNN_MINOR 5
#define CUDNN_PATCHLEVEL 1

 

 

다음 포스팅에서는 Anaconda 설치에 관한 내용을 소개하겠습니다.

Ubuntu에 Anaconda 설치