本地查看SSH远程连接服务器上的TensorBoard

本机系统Mac OS,用SSH远程连接服务器跑pytorch代码时需要用到Tensorboard,记录一下使用过程

安装TensorboardX:

pip install tensorboardX
pip install tensorboard

注意如果只使用TensorboardX,就没必要安装TensorFlow
在代码中添加Tensorboard的记录代码,参考
https://blog.csdn.net/bigbennyguo/article/details/87956434

此时在 uns文件夹下会生成相应的日志目录
新建一个SSH连接:
ssh -L 16006:127.0.0.1:6006 account@server.address

进入到刚刚 uns文件夹的根目录:
tensorboard --logdir runs
然后在本机访问 http://127.0.0.1:16006/ 即可

现在已经有官方支持的TensorBoard了:
https://pytorch.org/tutorials/recipes/recipes/tensorboard_with_pytorch.html

$ tensorboard dev upload --logdir runs 
--name "My latest experiment"  # optional
--description "Simple comparison of several hyperparameters" # optional

复制链接就可以远程查看

原文地址:https://www.cnblogs.com/limitlessun/p/14641370.html