《深度学习》TensorFlow之TensorBoard demo

import tensorflow.compat.v1 as tf

import os

os.environ["CUDA_VISIBLE_DEVICES"] = "-1"
tf.disable_v2_behavior()
a = tf.constant(4, name = "input_a")
b = tf.constant(2, name = "input_b")
c = tf.multiply(a, b, name = "mul_c")
d = tf.add(a, b, name = "add_d")
e = tf.add(c, d, name = "add_e")

sess = tf.Session()
print(sess.run(e))
write = tf.summary.FileWriter('./my_graph/1', sess.graph)

os.environ[“CUDA_VISIBLE_DEVICES”] = “-1” 是禁用GPU加速。

运行发现当前文件夹下多了个my_graph,里面有个1文件夹，1里面有日志数据。

运行TensorBoard的方法：

进入到保存日志的文件，运行语句：tensorboard --logdir=data （data是保存日志文件的全目录）

成功的话，最下面会显示一个地址，一般是http://localhost:6006/

复制到浏览器打开，就可以看到数据流图啦