win10查看Tensorflow是GPU还是CPU

打开终端(注:只有在终端中运行代码才会显示,在编译器上却不行)

win10查看Tensorflow是GPU还是CPU_第1张图片

输入以下代码

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。

win10查看Tensorflow是GPU还是CPU_第2张图片

win10查看Tensorflow是GPU还是CPU_第3张图片

你可能感兴趣的:(python,tensorflow,gpu)