Nvidia GPU Driver, CUDA, cuDNN 버전 확인(with. ubuntu)

Ubuntu 18.04 LTS에서 진행되어 작성된 포스팅입니다.

GPU Driver Version

• Driver Version: GPU 드라이버의 버전을 의미
• CUDA Version: 설치된 CUDA Version을 의미하는 것이 아닌 현재 GPU 드라이버 버전을 기준으로 가장 높은 CUDA Toolkit의 버전을 의미합니다.

CUDA Version

nvcc -V
nvcc -version

nvcc: NVIDIA (R) Cuda compiler driver
Copyright (c) 2005-2021 NVIDIA Corporation
Built on Sun_Feb_14_21:12:58_PST_2021
Cuda compilation tools, release 11.2, V11.2.152
Build cuda_11.2.r11.2/compiler.29618528_0

nvidia-smi의 cuda 버전은 11.4로 표기되는 반면, nvcc -V을 통해서 확인한 cuda 버전은 11.2로 표기되는 것을 알 수 있습니다. CUDA의 API는 Driver API, Runtime API 2가지의 API를 지니고 있습니다.

• Driver API: GPU driver installer에 의해 설치됨.
• Runtime API: CUDA toolkit installer에 의해 설치됨.

nvidia-smi로 보여주는 CUDA 버전의 경우 Driver API로부터 받은 CUDA 버전 정보이며,
nvcc로 보여주는 CUDA 버전의 경우 Runtime API로부터 받은 CUDA 버전 정보라고 합니다.

만약 GPU Driver 설치와 CUDA를 별도로 설치했다면, 이 둘의 버전이 다를 수 있다고 합니다.(참고)

cuDNN Version

# 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

저는 8.1.1 버전으로 확인이되네요. 아래 명령어를 통해서 확인했습니다.

#define CUDNN_MAJOR 8
#define CUDNN_MINOR 1
#define CUDNN_PATCHLEVEL 1
--
#define CUDNN_VERSION (CUDNN_MAJOR * 1000 + CUDNN_MINOR * 100 + CUDNN_PATCHLEVEL)

#endif /* CUDNN_VERSION_H */

Pytorch CUDA Toolkit Version

python 인터프리터를 직접 실행해서 확인보시면 됩니다.

>>> import torch
>>> torch.__version__
'1.12.1+cu116'