cuda에 맞는 cudnn 설치하기

재재

안녕하세요,
재재입니다.

cuda 를 설치하셨다면,
cuda에 맞는 cudnn 을 설치하셔야 할 시간입니다.

아직 설치하지 못하신 분들은,
아래 포스팅을 참고해주세요.

(1) cuda 에 맞는 cudnn 다운로드

먼저, 아래 페이지에서 설치하신 cuda 버전에 맞는 cudnn 을 찾으셔야 합니다.

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

반드시 Runtime Library 와 Developer Library 를 다운로드 합니다.

(2) cuda 11.0 기준 cudnn 설치

# setup your cudnn for cuda
sudo dpkg -i ~/Downloads/libcudnn8_8.1.0.77-1+cuda11.2_amd64.deb
sudo dpkg -i ~/Downloads/libcudnn8-dev_8.1.0.77-1+cuda11.2_amd64.deb

Downloads 폴더에 2개의 deb package 가 다운로드 되어있으셔야 합니다.