2024-11-21 一站式 AI 平台生日大派对!2024-11-21 一站式 AI 平台生日大派对! 无问芯穹特别推出多项超值福利!立即参与
Skip to content

TensorBoard

如果在 AIStudio 开发机内运行 TensorBoard,可从开发机详情页直接跳转 TensorBoard 数据看板。

前提条件

AIStudio 开发机实例中必须运行 TensorBoard 可视化看板工具。

请确保镜像中已包含 TensorBoard,或直接手动安装:

shell
python3 -m pip install tensorboard

启动 TensorBoard

在训练代码中,通过 SummaryWriter() 方法指定日志存储路径。启动 TensorBoard 可视化看板时,请指定该日志路径。

shell
# 以下命令直接在前台启动 TensorBoard 服务
# 在实际使用时建议放在后台,或使用 tmux 工具
tensorboard --logdir <训练日志储存路>

NOTE

TensorBoard 默认运行在 6006 端口。如使用 --port 指定为其他端口,则无法在 AIStudio 平台打开 TensorBoard 看板。

访问 TensorBoard 可视化看板

在开发机运行中,进入详情页,找到 TensorBoard,点击跳转,AIStudio 会打开 TensorBoard 看板。

alt text

常见问题

无法打开 TensorBoard 看板

请依次检查以下项目:

  1. 开发机是否在运行中。从 AIStudio 上访问 TensorBoard 依赖开发机内运行的 TensorBoard 服务。
  2. 是否已在开发机内部启动 TensorBoard 服务。您需要在 Shell 中自行启动 TensorBoard 服务。
  3. TensorBoard 服务是否运行在默认端口 6006。如果运行在非默认端口,AIStudio 无法自动转发开发机实例内的 TensorBoard 服务。

可以在本地设备上访问 TensorBoard 可视化界面吗?

并非只能在 AIStudio 上访问 TensorBoard 可视化界面。

TensorBoard 可视化看板是用户在开发机内自行运行的服务。AIStudio 平台为了方便用户访问 TensorBoard 可视化界面,转发了开发机实例内的 TensorBoard 可视化看板服务。

AIStudio 上访问 TensorBoard 要求登录 AIStudio 账号。如果想要从自己的设备上访问看板,可以使用 SSH 端口转发实现。详见公网访问开发机内 HTTP 服务

参考资源