tensorboardX在远程服务器上的安装与可视化

本文介绍了如何通过SSH隧道将远程服务器的tensorboard界面转发到本地,以便于在本地查看模型训练结果。步骤包括pip安装tensorboardX,设置SSH隧道,服务器端启动tensorboard,以及本地访问方法。
摘要由CSDN通过智能技术生成

1. 安装

  tensorboardX的安装非常简单,直接一行命令搞定:

pip install tensorboardX

2. 使用

  由于模型训练通常在远程服务器上进行,而服务器上没有可视化界面,无法直接查看tensorboard的结果,因此需要采用一些措施将服务器上的界面转发到本地,这里采用ssh端口转发的方式实现。
  首先借助MobaXterm实现ssh端口转发:
在这里插入图片描述

  1. 选择状态栏中的Tunneling,进入隧道设置界面,选择新建隧道:
    在这里插入图片描述
  2. 配置隧道端口并保存:
    在这里插入图片描述
    上图中,Remote server/Remote port依次填入localhost和6006(和使用tensorboard命令生成记录时的端口保持一致);SSH server/SSH login/SSH port分别填入远程服务器的IP/用户名/端口号;Forward port填入本地端口号6006(也可是其他可用端口,用于接收远程端口转发)
    以上就完成了ssh隧道的建立,将远程6006端口的内容转发到本地6006端口。

  然后在服务器上生成tensorboard记录:在tensorboard日志目录外执行命令:

tensorboard --logdir=${YOUR TFBORAD DIR} --port=6006

  最后在本地打开https://localhost:6006即可查看tensorboard的结果。

  • 3
    点赞
  • 12
    收藏
    觉得还不错? 一键收藏
  • 0
    评论
评论
添加红包

请填写红包祝福语或标题

红包个数最小为10个

红包金额最低5元

当前余额3.43前往充值 >
需支付:10.00
成就一亿技术人!
领取后你会自动成为博主和红包主的粉丝 规则
hope_wisdom
发出的红包
实付
使用余额支付
点击重新获取
扫码支付
钱包余额 0

抵扣说明:

1.余额是钱包充值的虚拟货币,按照1:1的比例进行支付金额的抵扣。
2.余额无法直接购买下载,可以购买VIP、付费专栏及课程。

余额充值