드레스룸 시스템행거 인테리어

CUDNN

GPU
2020. 6. 15. 15:39
반응형

CUDNN : CUDA Deep Neural Network library

cudnn은 엔비디아 cuda 딥 뉴럴 네트워크 라이브러리, 즉 딥 뉴럴 네트워크를 위한 GPU

가속화 라이브러리의 기초 요소로 컨볼루션[convolution], 풀링[pooling], 표준화[Nomarlization],

활성화[Activation]와 같은 일반적인 루틴을 빠르게 이행할 수 있도록 하는 라이브러리입니다.

 

cudnn 가속화 프레임워크

Tensorflow, Theano,Torch, Caffe등

 

 

CUDNN을 설치 방법

https://developer.nvidia.com/cudnn

 

NVIDIA cuDNN

NVIDIA cuDNN The NVIDIA CUDA Deep Neural Network library (cuDNN) is a GPU-accelerated library of primitives for deep neural networks. cuDNN provides highly tuned implementations for standard routines such as forward and backward convolution, pooling, norma

developer.nvidia.com

회원가입 및 로그인 후 

 

본인의 아키텍처에 맞는 cudnn version을 다운로드 후 사용하면 된다.

 

ex)

cudnn-10.0-linux-x64-v7.4.2.24.solitairetheme8

파일 형식이 정확히는 모르겠지만 tar로 압축풀기가 된다.

 

# tar xvf cudnn-10.0-linux-x64-v7.4.2.24.solitairetheme8

 

압축을 풀면 cuda폴더에 하위폴더 include, lib64가 생성 됨.

 

# cd /cuda

 

# cp inculde/cudnn.h /usr/local/cuda-10.0/include

  cuda version을 안써줘도 되지만 여러개의 cuda version을 사용하는 기업도 있다 

 

# cp lib64/libcudnn* /usr/local/lib64/cuda-10.0/lib64
# chmod a+r /usr/local/cuda-10.0/lib64/libcudnn*

  a 모든사용자 + 기존 권한에 권한 추가 r 읽기 권한

 

# cat /usr/local/cuda-10.0/incude/cudnn.h |grep CUDNN_MAJOR -A 2
#define CUDNN_MAJOR 7
#define CUDNN_MINOR 4
#define CUDNN_PATCHLEVEL 2
--
#define CUDNN_VERSION (CUDNN_MAJOR * 1000 + CUDNN_MINOR * 100 + CUDNN_PATCHLEVEL)

#include "driver_types.h"

  cudnn version minor 2자리까지 확인

 

 

이로써 간단하게 CUDNN을 설치하는 방법을 알아 보았다.

반응형

'GPU' 카테고리의 다른 글

Bazel 설치 방법  (0) 2020.06.18
CUDA 설치  (0) 2020.06.16
2080Ti-TF_benchmark_test  (0) 2020.06.14
NVML:Driver/library verion mismatch  (0) 2020.06.12
nvidia-driver설치  (0) 2020.06.12

이 글을 공유합시다

facebook twitter kakaoTalk kakaostory naver band