验证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版本

tensorflow2添加命令使用cpu训练

tensorflow如何设置只在cpu上运行

[tensorflow] tensorflow-cpu/gpu 安装过程

tensorflow怎么删除gpu版本,下载cpu版本

tensorflow gpu版本怎么换到cpu版本