验证tensorflow版本是GPU还是CPU
Posted lion-zheng
tags:
篇首语:本文由小常识网(cha138.com)小编为大家整理,主要介绍了验证tensorflow版本是GPU还是CPU相关的知识,希望对你有一定的参考价值。
reference: https://blog.csdn.net/zlase/article/details/79261348
import numpy
import tensorflow as tf
a = tf.constant([1.0, 2.0, 3.0, 4.0], shape=[2, 2], name=‘a‘)
b = tf.constant([1.0, 2.0, 3.0, 4.0], shape=[2, 2], name=‘b‘)
c = tf.matmul(a, b)
sess = tf.Session(config=tf.ConfigProto(log_device_placement=True))
print(sess.run(c))
以上是关于验证tensorflow版本是GPU还是CPU的主要内容,如果未能解决你的问题,请参考以下文章
教你如何一招查看 Tensorflow 是GPU版本 or CPU版本