- 查看显卡信息
lspci |grep -i vga
lspci -k | grep -A 2 -i "VGA"
- 查看驱动设备
ubuntu-drivers devices
- 查看cuda版本
nvidia-smi
nvcc --version
cat /usr/local/cuda/version.txt
- 查看cudnn版本
cat /usr/local/cuda/include/cudnn.h | grep CUDNN_MAJOR -A 2
- 验证tensorflow gpu是否可用.
import tensorflow as tf
sess = tf.Session(config=tf.ConfigProto(log_device_placement=True))
- 验证torch gpu是否可用
torch.cuda.is_available()
torch.backends.cudnn.enabled