服务器上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 服务器连接端口)

2. 在服务器上运行tensorboard

tensorboard --logdir=./runs

3. 在本机浏览器运行

http://127.0.0.1:6006