测试程序：

import tensorflow as tfwith 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 + bprint('结果是：%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 不打印日志，不然会刷屏