우분투와 친해지기
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
설치가 완료되었습니다.
반응형