本機系統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
復制鏈接就可以遠程查看