测试程序：

import tensorflow as tf

with tf.Session(config=tf.ConfigProto(allow_soft_placement=True, log_device_placement=False)) as sess:
    a = tf.constant(1)
    b = tf.constant(3)
    c = a + b
    print('结果是：%d\n 值为：%d' % (sess.run(c), sess.run(c)))

 

输出显卡型号、内存等：

tf.Session(config=tf.ConfigProto(allow_soft_placement=True, log_device_placement=False))

allow_soft_placement=True 表示使用不能使用显卡时使用cpu

log_device_placement=False 不打印日志，不然会刷屏