TensorBoard不需要额外安装,在TensorFlow安装时已自动完成。
Step 1:在原来的code中加入tf.reset_default_graph(), logdir以及writer.
import tensorflow as tf
tf.compat.v1.reset_default_graph() #清除default graph和不断增加的节点
input1 = tf.constant([1.0, 2.0, 3.0], name='input1')
input2 = tf.Variable(tf.random.uniform([3]), name='input2')
output = tf.add_n([input1, input2], name='add')
logdir='E:/python_learning' # logdir改为自己机器上的合适路径
writer = tf.compat.v1.summary.FileWriter(logdir,tf.get_default_graph())
# 生成一个写日志的writer,并将当前tensorflow计算图写入日志
writer.close()
Step 2:在Anaconda Prompt中先进入日志存放的目录(这里是E:\python_learning),再运行TensorBoard并将日志地址指向程序日志输出的地址(命令:tensorboard --logdir=E:\python_learning)
Step 3:打开浏览器,输入地址localhost:6006(启动服务端口默认为6006;使用--port参数可改编启动服务的端口)
TensorBoard常用API:
以上。参考Mooc课程是吴明晖老师的《深度学习应用开发-TensorFlow实践》。