[Cuda cudnn version check] #cuda #cudnn #nvidia

check_cuda_cudnn.md

To check nvidia driver

modinfo nvidia

To check cuda version

cat /usr/local/cuda/version.txt

nvcc  --version

To check cudnn version

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

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

To check GPU Card info

nvidia-smi

Python (Show what version of tensorflow in your PC.)

• for Python 2

python -c 'import tensorflow as tf; print(tf.__version__)'  

• for Python 3

python3 -c 'import tensorflow as tf; print(tf.__version__)'  

gpu check

CUDA_DEVICE_ORDER=PCI_BUS_ID CUDA_VISIBLE_DEVICES=1 python import pytorch torch.cuda.current_device()