[Tensorflow] Tensorflow, CUDA, CuDnn 버전 확인

방법 1

import tensorflow as tf
from tensorflow.python.platform import build_info as build
print(f"tensorflow version: {tf.__version__}")
print(f"Cuda Version: {build.build_info['cuda_version']}")
print(f"Cudnn version: {build.build_info['cudnn_version']}")

방법 2

CUDA 버전 확인

nvcc -V

• nvcc -V명령어가 안되면 .bashrc에 환경변수 추가해줘야 함

  export 

  nvcc없을 때 CUDA 버전 확인

  /usr/local/cuda-버전/bin/nvcc -V

CUDNN 8 이전 버전 확인

cat /usr/include/x86_64-linux-gnu/cudnn_v*.h | grep CUDNN_MAJOR -A 2

CUDNN 8 이후 버전 확인

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

또는

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