如何在 Colaboratory 中使用 Tensorboard

Posted 2023-04-18

【中文标题】如何在 Colaboratory 中使用 Tensorboard

【英文标题】：How to use Tensorboard in Colaboratory

【发布时间】：2018-07-02 15:08:01

【问题描述】：

是否可以在 Colaboratory 中使用 Tensorboard。在本地运行 tensorboard 会显示有关模型行为（如损失等）的丰富信息。与 Colaboratory 合作时是否可以获得相同的信息（https://colab.research.google.com）。

【问题讨论】：

我一直在想同样的事情（加上我有一些我感兴趣的库可以产生 tensorboard 调试）但是，一个不完美但简单的解决方法可能是下载日志文件夹并在本地运行 tensorboard。

Can I use Tensorboard with Google Colab?的可能重复

【参考方案1】：

您有两种选择，您可以使用其中一种 Python 程序，该程序允许您通过隧道连接到托管您的 Python 应用程序的机器实例。我测试了这个： https://github.com/taomanwai/tensorboardcolab

!pip install -U tensorboardcolab
from tensorboardcolab import *
import shutil

#clean out the directory
shutil.rmtree('./Graph', ignore_errors=True)
os.mkdir('./Graph')

tf.reset_default_graph()

#will start the tunneling and will print out a link:
tbc=TensorBoardColab()

#**here you construct your model**

sess = tf.Session()
output = sess.run(....)
sess.close()

train_writer = tbc.get_writer();
train_writer.add_graph(sess.graph)

train_writer.flush();
tbc.close()

另一种解决方案是将所有文件压缩并下载到您的计算机上。

【参考方案2】：

现在您无需 ngrok 和其他软件包即可在 Colab 中使用 Tensorboard：

import os
logs_base_dir = "tb_runs"
os.makedirs(logs_base_dir, exist_ok=True)
%load_ext tensorboard
%tensorboard --logdir logs_base_dir
# Now Tensorboard interface appear in this cell output

官方示例：https://www.tensorflow.org/tensorboard/get_started