pytorch + tensorboard远程可视化

安装

pip install tensorboard

使用

流程就是在代码中把值写入log文件，然后用tensorboard读取就行

from torch.utils.tensorboard import SummaryWriter   
import torch
import numpy as np
writer = SummaryWriter(logsdir)
x = np.arange(0, 5, 0.1)
x = torch.from_numpy(x)

for i in x:
    yi = torch.exp(i)
    writer.add_scalar('exp', yi, i)

注意SummaryWriter(logsdir) 其中的logsdir是要保存的路径， writer.add_scalar(tag, scalar_value, global_step=None, walltime=None)这里的tag指定变量名字，scalar_value是变量的值，global_step为x轴坐标。

保存如下

配置

1. 在本地计算机上使用SSH 并将远程服务器的端口映射到本地的计算机。

ssh -L 6006:127.0.0.1:6006 username@server_ip -p server_port

2. 在远程服务器开启tensorboard

tensorboard --logdir logsdir --port=6006

这里的logsdir 就是上文的保存路径

运行后弹出如下链接

粘贴到本地服务器即可看到