modinfo nvidia
cat /usr/local/cuda/version.txt
nvcc --version
cat /usr/local/cuda/include/cudnn.h | grep CUDNN_MAJOR -A 2
cat /usr/include/cudnn.h | grep CUDNN_MAJOR -A 2
nvidia-smi
- for Python 2
python -c 'import tensorflow as tf; print(tf.__version__)'
- for Python 3
python3 -c 'import tensorflow as tf; print(tf.__version__)'
CUDA_DEVICE_ORDER=PCI_BUS_ID CUDA_VISIBLE_DEVICES=1 python import pytorch torch.cuda.current_device()