Ubuntu深度学习环境配置常用命令

1. 查看显卡信息

lspci |grep -i vga
lspci -k | grep -A 2 -i "VGA"

2. 查看驱动设备

ubuntu-drivers devices

3. 查看cuda版本

cat /usr/local/cuda/version.txt

4. 查看cudnn版本

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

5. 验证tensorflow gpu是否可用.

import tensorflow as tf
sess = tf.Session(config=tf.ConfigProto(log_device_placement=True))

6. 验证torch gpu是否可用

torch.cuda.is_available()
torch.backends.cudnn.enabled