TensorBoard
如果在 AIStudio 开发机内运行 TensorBoard,可从开发机详情页直接跳转 TensorBoard 数据看板。
前提条件
AIStudio 开发机实例中必须运行 TensorBoard 可视化看板工具。
请确保镜像中已包含 TensorBoard,或直接手动安装:
shell
python3 -m pip install tensorboard
启动 TensorBoard
在训练代码中,通过 SummaryWriter()
方法指定日志存储路径。启动 TensorBoard 可视化看板时,请指定该日志路径。
shell
# 以下命令直接在前台启动 TensorBoard 服务
# 在实际使用时建议放在后台,或使用 tmux 工具
tensorboard --host 0.0.0.0 --logdir <训练日志储存路径>
NOTE
- 在启动 TensorBoard 服务时,请绑定到所有网络接口(
0.0.0.0
),否则无法预览。 - TensorBoard 默认运行在 6006 端口。如使用
--port
指定为其他端口,则开发机详情页的 Tensorboard 按钮失效。
访问 TensorBoard 可视化看板
在开发机运行中,进入详情页,找到 TensorBoard,点击跳转,AIStudio 会打开 TensorBoard 看板。
常见问题
无法打开 TensorBoard 看板
请依次检查以下项目:
- 开发机是否在运行中。从 AIStudio 上访问 TensorBoard 依赖开发机内运行的 TensorBoard 服务。
- 是否已在开发机内部启动 TensorBoard 服务。您需要在 Shell 中自行启动 TensorBoard 服务。
- TensorBoard 服务是否运行在默认端口
6006
。如果运行在非默认端口,AIStudio 无法自动转发开发机实例内的 TensorBoard 服务。
可以在本地设备上访问 TensorBoard 可视化界面吗?
并非只能在 AIStudio 上访问 TensorBoard 可视化界面。
TensorBoard 可视化看板是用户在开发机内自行运行的服务。AIStudio 平台为了方便用户访问 TensorBoard 可视化界面,转发了开发机实例内的 TensorBoard 可视化看板服务。
AIStudio 上访问 TensorBoard 要求登录 AIStudio 账号。如果想要从自己的设备上访问看板,可以使用 SSH 端口转发实现。
📖 SSH 端口转发:公网访问开发机内 HTTP 服务