[딥러닝] (CUDA 설치 후) cuDNN 설치하기

연산량이 많은 딥러닝을 위해 GPU를 사용하려면

CUDA 및 cuDNN 설치가 필요하다.

 

CUDA는 사용하려는 딥러닝 프레임워크 버전과 그래픽 드라이버 종류에 따라

CUDA Tool kit을 다운받아 설치할 수 있다.

 tensorflow-gpu 버전 확인, Pytorch 버전 호환, NVIDIA 그래픽 드라이버 호환

 

https://www.nvidia.co.kr/Download/index.aspx?lang=kr

Download the latest official NVIDIA drivers

https://developer.nvidia.com/cuda-toolkit-archive

CUDA Toolkit Archive

 

 

 

CUDA는 이미 설치되었다고 가정하고

cuDNN(old version)을 설치하는 방법을 설명한다.

 

---

 

1. cuDNN 압축파일 다운로드

 

아래 사이트에서 설치된 CUDA 버전과 호환되는 cuDNN 파일을 다운로드한다.

( NVIDIA 사이트에 회원가입이 되어 있어야 로그인 후 파일을 다운받을 수 있다. )

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

 

 

2. 압축 풀기

 

다운받은 cuDNN 압축파일을 푼다.

$ sudo tar -xzvf cudnn-9.0-linux-x64-v7.0.tgz

 

 

3. cuDNN 설치

 

압축을 풀면 cuda 폴더가 생성되는데, 생성된 폴더로 들어간다.

$ cd cuda

 

그리고 그 안의 파일들을

CUDA를 설치하면서 생긴 폴더 /usr/local/cuda-x.x 밑으로 복사한다.

(참고로 /usr/local/cuda-x.x 처럼 cuda 폴더 뒤에 버전 이름을 적어두면

나중에 여러 cuda 버전을 바꿔서 사용할 때 편하다.)

$ sudo cp include/cudnn.h /usr/local/cuda-x.x/include
$ sudo cp lib64/libcudnn* /usr/local/cuda-x.x/lib64

$ sudo chmod a+r /usr/local/cuda-x.x/lib64/libcudnn*

 

 

4. 설치 확인

 

cat 명령어 입력 후

아래와 같은 문구가 뜨면 설치에 성공한 것이다.

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

#define CUDNN_MAJOR 7
#define CUDNN_MINOR 6
#define CUDNN_PATCHLEVEL 1
--
#define CUDNN_VERSION (CUDNN_MAJOR * 1000 + CUDNN_MINOR * 100 + CUDNN_PATCHLEVEL)