Linux, Ubuntu CUDA와 cuDNN 버전을 확인하는 방법

Linux, Ubuntu CUDA와 cuDNN 버전을 확인하는 방법

 

1. CUDA 버전 확인:

• 터미널에서 CUDA 버전 확인: 터미널을 열고 다음 명령어를 입력하세요:

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 CUDA Compiler (nvcc)의 버전을 출력합니다. 여기서 CUDA Toolkit의 버전을 확인할 수 있습니다.
• 환경 변수를 통해 확인: 일부 시스템에서는 CUDA 설치 경로가 PATH 환경 변수에 추가되어 있습니다. 이 경로를 통해 CUDA 버전을 확인할 수 있습니다.

2. cuDNN 버전 확인:

• cuDNN은 CUDA와 함께 사용되는 라이브러리이며, 일반적으로 특정 경로에 설치됩니다 (예: /usr/local/cuda/include/ 또는 /usr/include/).
• cuDNN 버전을 확인하려면, cuDNN 헤더 파일을 찾아 해당 파일의 내용을 확인합니다. 다음 명령어를 사용할 수 있습니다:

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

 

결과

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

 

• 이 명령어는 cuDNN 버전을 나타내는 매크로가 정의된 cudnn_version.h 파일의 일부를 출력합니다.
• 경로가 /usr/local/cuda/가 아닌 경우, 시스템에 따라 다른 경로에 설치되었을 수 있습니다. 해당 경로를 찾아 위의 명령어를 적절히 수정해 사용하십시오.