本机系统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
复制链接就可以远程查看