1. Nvidia Driver 설치 여부 및 버전 확인

$ cat /proc/driver/nvidia/version

 정상적으로 설치되어 있는 경우 아래와 같이 출력된다.

NVRM version: NVIDIA UNIX x86_64 Kernel Module  430.34  Wed Jun 26 12:19:48 CDT 2019
GCC version:  gcc version 7.4.0 (Ubuntu 7.4.0-1ubuntu1~18.04.1) 

 

이외 nvidia-smi를 이용해서 확인하는 방법도 있다. (이것을 더 추천한다.)

$ nvidia-smi

 

2.  CUDA

$ nvcc --version

 정상적으로 설치되어 있는 경우 아래와 같이 출력된다.

nvcc: NVIDIA (R) Cuda compiler driver
Copyright (c) 2005-2019 NVIDIA Corporation
Built on Sun_Jul_28_19:07:16_PDT_2019
Cuda compilation tools, release 10.1, V10.1.243

 

nvidia-smi는 수행되는데 nvcc 가 수행 안될 경우, 

보통은 path 설정 문제이다. 하기와 같이 .bashrc 파일에 경로(path)를 추가한다.

# VI 에디터로 .bashrc 파일에 경로 추가

$ vi ~/.bashrc

export PATH=/usr/local/cuda/bin:$PATH
export LD_LIBRARY_PATH=/usr/local/cuda/lib64:$LD_LIBRARY_PATH

 

# .bashrc 재수행

$ source ~/.bashrc

 

3. CUDNN

버전별로 스크립트가 하기와 같이 다르다.

# cuDNN 8.x.x 이전 버전
cat /usr/local/cuda/include/cudnn.h | grep CUDNN_MAJOR -A 2

# cuDNN 8.x.x 이후 버전
cat /usr/local/cuda/include/cudnn_version.h | grep CUDNN_MAJOR -A 2

 

 

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

→ 정상적으로 설치되어 있는 경우 아래와 같이 출력된다.

#define CUDNN_MAJOR 7
#define CUDNN_MINOR 6
#define CUDNN_PATCHLEVEL 3
--
#define CUDNN_VERSION (CUDNN_MAJOR * 1000 + CUDNN_MINOR * 100 + CUDNN_PATCHLEVEL)

#include "driver_types.h"

 

반응형

+ Recent posts