본문 바로가기

Tect.tip/Linux

[install]cudnn 최신버전 설치

구성 환경

Ubuntu 18.04.4 LTS

RTX 2080 Ti

Cudnn 이란?

cuDNN(Deep Neural Network)은 GPU-accelerated 라이브러리로 딥러닝을 돌릴 때 사용합니다. cuDNN을 on/off 했을 때 속도 차이가 넘사벽 수준입니다. 선택이 아니라 필수로 사용해야 하는 라이브러리죠.

Cudnn 홈페이지에서 원하는 버전을 찾기

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

로그인을 하고 설문조사를 간단히 마친 후, cuDNN 7.6.5을 다운로드 cuda 10.2 (.tar 파일)

$ tar -xvf cudnn-10.2-linux-x64-v7.6.5.32.tgz

압축을 풀면 cuda라는 폴더가 생기고, cuda 폴더 안에는 헤더파일이 들어있는 include/ 폴더와 라이브러리가 들어있는 lib64/ 폴더가 보인다.

include/ 내 헤더 파일 cudnn.hlib64/ 내 라이브러리 파일을 cuda 설치 폴더에 각각 복사하면 된다.

#cudnn.h 파일을 복사합니다.

$ sudo cp ./cuda/include/* /usr/local/cuda-10.2/include

#라이브러리 파일 복사 시, 옵션 -P를 붙여서 심볼릭 링크가 깨지지 않도록 합니다.

$ sudo cp -P ./cuda/lib64/* /usr/local/cuda-10.2/lib64

#모든 사용자에게 라이브러리 실행권한을 부여합니다.

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

Cudnn 확인

$ nv