nvdia driver, CUDA, CUDNN 설치 여부 및 버전 검사

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

$ cat /usr/local/cuda/include/cudnn.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"