win10查看Tensorflow是GPU还是CPU

打开终端（注：只有在终端中运行代码才会显示，在编译器上却不行）

输入以下代码

import tensorflow as tf
a = tf.constant([1.0, 2.0, 3.0, 4.0, 5.0, 6.0], shape=[2, 3], name='a')
b = tf.constant([1.0, 2.0, 3.0, 4.0, 5.0, 6.0], shape=[3, 2], name='b')
c = tf.matmul(a, b)
sess = tf.Session(config=tf.ConfigProto(log_device_placement=True))
print (sess.run(c))

运行结果如下图，我的显示是GPU。