[Linux] CUDA 11와 함께 쓰는 cuDNN 8 설치하기

* CUDA 설치

2021.03.25 - [리뷰/Ubuntu 20.04] - [Linux] NVIDIA RTX 3090을 위한 CUDA 11 설치 방법

 

---

<cuDNN 8 설치>

 

설치 과정을 간략하게 설명하자면,

공식 사이트에서 파일 다운로드 → 압축 해제 → 디렉토리 이동 → 권한변경

* 공식 사이트에서 파일을 다운로드하기 위해 NVIDIA 로그인을 해야 됨

cuDNN 설치 공식 사이트: https://developer.nvidia.com/cudnn

동의한 후에 밑줄 클릭하면 아래와 같은 화면이 나옵니다.

위와 같이 나왔다면 빨간색으로 체크한 링크를 눌러 파일을 다운로드합니다.

 

'?'는 여러분이 다운로드한 파일명과 똑같이 해주시면 됩니다.

$ tar -xzvf cudnn-x.x-linux-x64-v8.x.x.x.tgz

압축 해제한 경로에서부터 cuda 디렉토리로 이동시켜준 후에 사용 권한을 변경합시다.

$ sudo cp cuda/include/cudnn*.h /usr/local/cuda/include
$ sudo cp cuda/lib64/libcudnn* /usr/local/cuda/lib64
$ sudo chmod a+r /usr/local/cuda/include/cudnn*.h /usr/local/cuda/lib64/libcudnn*

위의 과정까지 따라왔다면 마지막은 역시 버전 확인이 필요합니다.

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

cuDNN 8 버전 확인

파랑 밑줄친 곳에 8이 있네요 → 매우 잘 설치됨