CUDNN 설치법

CUDA를 설치했으니 이번에는 CUDNN 을 설치하려고한다.

사실 처음엔 CUDA도 잘모르는데 CUDNN까지 설치하려고하니까 막상 무서웠는데 한두번 해보니까 별거아니었다. 

바로시작해보자!

 

먼저 CUDA와 비슷하게 CUDNN 아카이브에서 CUDA 버전에 맞고 자기가원하는 CUDNN 버전을 설치해준다

나는 tgz 파일로 받아서 압축해제하고 수동으로 설치하였다..

 

먼저 해당링크에 접속한다

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

cuDNN Archive

여기에 접속을하면 로그인을 해야 다운로드 받을수있다. 귀찮지만 앞으로를 위해서 NVIDIA Developer는 가입을 하길 권장한다.

 

나는 Ubuntu 환경에서 설치하려고했으므로 

Linux용 cuDNN 라이브러리(x86 64)를 다운로드 받았다.

 

압축파일을 다운로드 한후 편한 경로에 압축파일을 풀어준다

 

tar -xvzf cudnn-11.0-linux-x64-v8.0.5.39.tgz

그러면 cudnn8.0.5 이런식으로 디렉토리가 생성이 된다.

여기로 먼저 들어가보자

 

cd cudnn8.0.5/cuda/

여기로 접속하면 include 라는 디렉토리와 lib64 라는 디렉토리가 보일것이다.

여기서 include디렉토리안에있는 파일과 lib64디렉토리안에있는 파일을 cuda 가 설치되어있는곳으로 옮겨줄거다.

 

그방법은 다음과 같다.

root@1234aaa123:/data/cudnn8.0.5/cuda/include# cp * /usr/local/cuda/include
root@1234aaa123:/data/cudnn8.0.5/cuda/lib64# cp * /usr/local/cuda/lib64

#뒤에있는 코드를 실행 시켜주기를 바란다

 

이러고 난 후 이제 CUDNN이 정상적으로 설치되어있는지 확인해보자

 

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

그럼 다음과같이 버전이 확인될것이다

#define CUDNN_MAJOR 8
#define CUDNN_MINOR 0
#define CUDNN_PATCHLEVEL 5

이러면 설치완료된것이다. 

 

다음에는 NCCL에 설치하는것에 대해서도 알아보자!