服务器上tensorboard的使用

环境:
ubuntu服务器、pytorch、python2.7

安装:

  1. pip install tensorboardX
  2. pip install tensorflow

代码中调用tensorboardX

import numpy as np
from torch.utils.tensorboard import SummaryWriter

writer = SummaryWriter(comment='test_tensorboard')

for x in range(100):
    writer.add_scalar('y=2x', x * 2, x)
    writer.add_scalar('y=pow(2, x)', 2 ** x, x)

    writer.add_scalars('data/scalar_group', {"xsinx": x * np.sin(x),
                                             "xcosx": x * np.cos(x),
                                             "arctanx": np.arctan(x)}, x)
writer.close()

在服务器启动tensorboardX,并在本地浏览器中显示

  1. 首先使用ssh登陆服务器并转发端口
ssh -L 6006:127.0.0.1:6006 远程服务器用户名@远程服务器IP -p 服务器连接端口
(ssh -L 本地端口:本地IP:远程端口 远程服务器用户名@远程服务器IP -p 服务器连接端口)
  1. 在服务器上运行tensorboard
tensorboard --logdir=./runs
  1. 在本机浏览器运行
http://127.0.0.1:6006

在这里插入图片描述

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

请填写红包祝福语或标题

红包个数最小为10个

红包金额最低5元

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

抵扣说明:

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

余额充值