cuDNN(CUDA Deep Nural Network Library) 설치하기

cuDNN이란, cuda 딥러닝과 관련한 라이브러리로서 딥러닝과 관련한 오픈 소스들이 위 라이브러리를 활용해서 작동합니다.

즉, 우리가 stable diffusion을 활용하여 이미지 생성 작업을 할 때, cuDNN도 설치해 줘야 관련한 확장 프로그램 및 버전 업데이트 등을 진행했을 때 문제없이 stable diffusion이 작동할 수 있습니다.

 

그럼 이제 설치하는 방법을 알아볼까요. (ubuntu 22.04 기준)

아래 사이트에서 다운로드가 가능합니다.

로그인을 하여야 접근 가능한데, 간단하게 회원 가입을 하면 사용 가능합니다.

 

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

Log in

반응형

회원 가입을 완료한 후 로그인을 하면, 다운로드를 받을 수 있습니다.

download 페이지 화면. 아래의 빨간 박스 친 Archived cuDNN Releases 를 클릭해주자

 

"Archived cuDNN Releases" 를 클릭하면 버전 별로 다양한 cuDNN을 확인할 수 있습니다.

본인의 CUDA 버전에 맞는 cuDNN을 확인하여 설치를 진행합니다.

 

여기서는 8.6.0 버전을 예시로 하여 설치를 진행해 보겠습니다.

(각자 본인 환경에 맞춰서 버전 설치를 진행하면 됩니다. 그 중 최신 버전 설치를 권장합니다.)

 

해당 버전을 클릭하면 여러가지 설치 옵션이 나오는데, 해당하는 os 환경(여기서는 ubuntu) 중에서 반드시, "Linux x86_64 (Tar)"버전으로 다운로드 합니다.

 

다운로드가 완료되었다면, 터미널 창을 켜고 다운로드 경로로 이동한 다음, 다음의 명령어를 입력합니다.

 

* 8.6.0 버전의 예시입니다. 만약의 다른 버전을 사용하신다면 숫자 부분을 그에 맞게 수정해서 사용해 주세요

* 정확한 버전명은 다운로드받은 파일 명에서 확인할 수 있습니다.

* 예를 들어 8.9.6 버전의 경우, 'cudnn-linux-x86_64-8.9.6.50_cuda11-archive.tar' 파일 명을 가지며, 8.9.6.50 버전임을 알 수 있습니다.

# 다운로드받은 압축 파일을 풀고, /usr/local/cuda 디렉토리로 복사하는 명령어
tar -xvf cudnn-linux-x86_64-8.6.0.163_cuda11-archive.tar.xz
cd cudnn-linux-x86_64-8.6.0.163_cuda11-archive
sudo cp include/cudnn* /usr/local/cuda/include
sudo cp lib/libcudnn* /usr/local/cuda/lib64
sudo chmod a+r /usr/local/cuda/include/cudnn.h /usr/local/cuda/lib64/libcudnn*

 

위 과정을 진행하면, 설치가 완료됩니다.

정상적으로 설치되었는지 확인하기 위해 설치된 cuDNN의 버전을 확인합니다.

# 터미널 창에서 아래의 명령어를 입력
cat /usr/local/cuda/include/cudnn_version.h | grep CUDNN_MAJOR -A 2

 

 

정상적으로 설치되었다면, 아래와 같은 형식으로 버전 정보가 출력됩니다.(아래는 8.6.0 예시)

8.6.0 버전이 정상적으로 설치된 화면(버전 확인)