[Ubuntu 20.04 LTS] CUDNN local설치

Nvidia-driver와 CUDA 설치를 완료 했으면 CUDNN을 설치해야 한다. 

아래의 사이트에 들어가서 CUDA 버전에 맞는 CUDNN버전을 찾는다. 

nvidia 계정이 필요하고 이 과정은 생략한다.  

https://developer.nvidia.com/cudnn

CUDA Deep Neural Network

CUDA 버전에 맞는 CUDNN을 찾으면 다운로드를 받는다. 

다운로드 받은 위치에서 압축을 푼다. 

$ tar zxvf [cudnn 파일]

 

압축이 풀리면 아래 3가지의 명령어를 입력한다. 

$ sudo cp cuda/include/cudnn*.h /usr/local/cuda/include
$ sudo cp -P cuda/lib64/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

CUDNN 설치 확인

---

Reference 

https://www.youtube.com/watch?v=mmp-hay0eeM&t=8s

---

정확한 정보 전달보단 공부 겸 기록에 초점을 둔 글입니다. 틀린 내용이 있을 수 있습니다.
틀린 내용이나 다른 문제가 있으면 댓글 남겨주시거나 또는 이메일로 보내주시면

감사하겠습니다.