JupyterLab TensorBoard Pro：功能完备的 TensorBoard 插件

研发背景

在萤火 JupyterLab 中使用

创建实例

1. 从 launcher 面板创建2. 通过快捷命令创建

设置参数

管理实例

在本地安装使用

总结

JupyterLab TensorBoard Pro：功能完备的 TensorBoard 插件

Aircloud    August 01, 2023

对于深度学习开发者和研究者们来说，TensorBoard 是帮助理解深度学习运行机理，训练可视化和调试的重要工具。

最近，幻方 AI 开发并集成了 JupyterLab TensorBoard 插件（JupyterLab TensorBoard Pro），帮助大家在使用幻方萤火集群训练模型的同时，也可以方便快捷的使用 Tensorboard 工具进行深度学习训练的调试。

该插件更新修复了社区里原插件的若干问题，并对性能进行了提升。我们对该插件进行了开源，方便用户在没有幻方萤火集群的 JupyterLab 环境的情况下，仍然可以下载和使用。下载安装方式如下

pip install jupyterlab-tensorboard-pro

开源地址：https://github.com/HFAiLab/jupyterlab_tensorboard_pro

研发背景

目前深度学习社区里面已经有了 jupyterlab_tensorboard（前端插件）和 jupyter_tensorboard（对应的后端插件）两个开源插件，然而这两个仓库已经很久没有更新迭代，对于一些外部提交的PR也没有及时合入（实际上，当我们发布 jupyterlab_tensorboard_pro 之后，jupyterlab_tensorboard 合入了一些社区 PR，但 jupyter_tensorboard 仍然存在无法配合较新的 tensorboard 版本使用的问题），基于此我们判断该项目作者已经不在积极地维护对应的仓库。

同时，现有社区的 TensorBoard 插件存在一定的体验问题：

需要同时安装两个 Python 包，并且需要版本对应，增加了使用和维护的成本；社区版本插件启动 TensorBoard 后端会持续 reload（从磁盘读取文件并更新），这会对 JupyterLab 的体验产生副作用，如果使用了分布式文件系统或者公用集群，甚至还会影响到其他用户。目前社区版本插件无法通过配置来禁用。功能体验不够完善，例如点击 TensorBoard 按钮之后会阻塞请求导致 JupyterLab 没有响应，并且没有任何提示，同时也对已经创建的 TensorBoard 实例的可视化管理功能也比较缺乏。

基于以上原因，幻方AI基于现有的社区版本二次开发了 TensorBoard 插件，以解决上述问题。在内部使用一段时间后，插件运行稳定并持续获得正向反馈，因此我们将插件开源以帮助