우분투와 친해지기

Linux Ubuntu 18.04 CuDNN 8.0.3 for CUDA 10.2 설치방법

로밀리언 2021. 1. 4. 22:42
반응형

Linux Ubuntu  18.04 CuDNN 8.0.3  for CUDA 10.2  설치방법 

 

developer.nvidia.com/rdp/cudnn-archive

 

cuDNN Archive

NVIDIA cuDNN is a GPU-accelerated library of primitives for deep neural networks.

developer.nvidia.com

사이트로 이동해서 로그인을 해줍니다.  cuDNN을 다운로드하기 위해서는 nvidia 계정이 필요합니다. 없으신 분들은 가입한 후 다운로드하시면 됩니다.

 

cuDNN Library for Linux [x86]을 클릭하면  .tgz파일을 다운로드할 수 있습니다.

 

tar -xvf cudnn-10.2-linux-x64-v8.0.3.33.tgz

다운로드한 폴더로 이동해서 압축을 풀어줍니다.

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

cuDNN header 파일들을  local/cuda/cuda-10.2/include 폴더에 붙여 넣어주는 명령어입니다.

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

cuDNN Library 파일들을  local/cuda/cuda-10.2/include 폴더에 붙여 넣어주는 명령어입니다.

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

라이브러리 실행 권한을 부여합니다.

 

 

cuDNN 8.x 이상 부터는 cudnn_version.h 파일에서 버전을 확인해야 하므로 

아래의 명령어로 정상적으로 설치되었는지 확인하 실 수 있습니다.

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

설치가 완료되었습니다.

반응형