本地查看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

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

进入到刚刚\runs文件夹的根目录:
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

复制链接就可以远程查看

posted @ 2021-04-10 16:44  Limitlessun  阅读(584)  评论(0编辑  收藏  举报