CUDA, cuDNN 설치 및 버전 확인

CUDA와 CUDnn은 운영체제 및 버전에 맞게 엔비디아 사이트에서 다운로드 받고 설치해야 한다.

• CUDA archive download
• cuDNN download
  

만약, 아나콘다로 설치하려면 다음 명령을 입력한다.

conda install cudatoolkit
conda install cudnn

설치 후 cudnn.h 파일 중 아래 부분을 수정한다.

#include "driver_types.h"   >   #include <driver_types.h>

아래와 같이 쉘 파일을 편집해 경로를 추가한다.

gedit ~/.bashrc

export PATH="/home/ktw/anaconda3/bin:$PATH"
export PATH="/usr/local/cuda-10.0/bin:$PATH"
export PATH="/usr/local/cuda-10.0/lib64:$PATH"
export LD_LIBRARY_PATH="/usr/local/cuda-10.0/lib64:$LD_LIBRARY_PATH"
export LIBRARY_PATH="/usr/local/cuda-10.0/lib64:$LIBRARY_PATH"

환경변수를 체크한다.

printenv

제대로 설치되었는 지 버전 등 체크한다.

검색 설

cd /usr/local/cuda-xx.x/lib64

ls

cat /usr/local/cuda/version.txt
nvcc  --version
cat /usr/local/cuda/include/cudnn.h | grep CUDNN_MAJOR -A 2

예제를 아래 폴더에서 home 폴더로 복사한다.

/usr/src/cudnn_sample_v7

cudnn 폴더 아래 mnist 예제 폴더안에서 다음 같이 메이크한다.

make

빌드된 결과를 실행하면 다음 결과를 얻을 수 있다.

레퍼런스

• cuda & cudnn installation