环境:
ubuntu服务器、pytorch、python2.7
安装:
- pip install tensorboardX
- 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,并在本地浏览器中显示
- 首先使用ssh登陆服务器并转发端口
ssh -L 6006:127.0.0.1:6006 远程服务器用户名@远程服务器IP -p 服务器连接端口
(ssh -L 本地端口:本地IP:远程端口 远程服务器用户名@远程服务器IP -p 服务器连接端口)
- 在服务器上运行tensorboard
tensorboard --logdir=./runs
- 在本机浏览器运行
http://127.0.0.1:6006