如何使用Tensorboard查看远程服务器的训练情况(pytorch也可以使用)

Tensorboard 在本机可以方便使用，但使用服务器时需要设置一下。

原理

建立ssh隧道，实现远程端口到本地端口的转发 具体来说就是将远程服务器的6006端口（tensorboard默认将数据放在6006端口）转发到本地的16006端口，在本地对16006端口的访问即是对远程6006端口的访问，当然，转发到本地某一端口不是限定的，可自由选择。

Mac或Linux系统

在登录远程服务器的时候使用命令：

ssh -L 16006:127.0.0.1:6006 [email protected]

代替一般ssh远程登录命令：ssh [email protected]。ip地址：127.0.0.1指代本地主机。

训练完模型之后使用如下命令：

tensorboard --logdir="/path/to/log-directory" --port=6006

其中，/path/to/log-directory为自己设定的日志存放路径，因人而异
最后，在本地浏览器访问地址：http://127.0.0.1:16006/

 

参考博客：https://blog.csdn.net/aiynmimi/article/details/88531695