기계는 거짓말하지 않는다

Ubuntu cuDNN 설치 본문

Linux

Ubuntu cuDNN 설치

KillinTime 2023. 3. 25. 22:31

cuDNN 설치 방법이다.

NVIDIA cuDNN Archive

 

cuDNN Archive

Download releases from the GPU-accelerated primitive library for deep neural networks.

developer.nvidia.com

다운로드는 NVDIA 계정이 있어야 한다.

CUDA Toolkit 설치 후, 먼저 위 NVIDIA 사이트에서 CUDA 버전과 맞는

cuDNN 버전을 선택하고 맞는 Linux 버전을 선택한다. (ex: cuDNN Library for Linux (x86_64))

 

tar 파일을 압축 해제하면 cuda 디렉터리가 생기는데 아래의 명령어를 입력한다.

# (version)에는 설치된 CUDA 버전을 입력한다. ex) cuda-11.4
sudo cp ./cuda/include/* /usr/local/cuda-(version)/include
sudo cp -P ./cuda/lib64/* /usr/local/cuda-(version)/lib64
sudo chmod a+r /usr/local/cuda-(version)/libcudnn*

예시

 

Comments