Tensorboard 在本機可以方便使用,但使用服務(wù)器時需要設(shè)置一下。
(一)windows系統(tǒng)
在Windows系統(tǒng)裝一個Xshell,在文件->屬性->ssh->隧道->添加,類型local,源主機填127.0.0.1(意思是本機),端口設(shè)置一個,比如12345,目標主機為服務(wù)器,目標端口一般是6006,如果6006被占了可以改為其他端口。
在服務(wù)器上運行 tensorboard --logdir='logs' --port =6006
在本機打開網(wǎng)頁127.0.0.1:12345
即可查看遠程的tensorboard。
如果用到了跳板機,上述的設(shè)置仍舊適用。通過跳板機連接遠程計算機,請參考 :https://blog.csdn.net/qq_36866808/article/details/79354366
(二)Mac或Linux系統(tǒng)
在登錄遠程服務(wù)器的時候使用命令:
ssh -L 16006:127.0.0.1:6006 account@server.address
(代替一般ssh遠程登錄命令:ssh account@server.address)
訓(xùn)練完模型之后使用如下命令:
tensorboard --logdir="/path/to/log-directory"
(其中,/path/to/log-directory為自己設(shè)定的日志存放路徑,因人而異)
最后,在本地訪問地址:http://127.0.0.1:16006/
原理
建立ssh隧道,實現(xiàn)遠程端口到本地端口的轉(zhuǎn)發(fā) 具體來說就是將遠程服務(wù)器的6006端口(tensorboard默認將數(shù)據(jù)放在6006端口)轉(zhuǎn)發(fā)到本地的16006端口,在本地對16006端口的訪問即是對遠程6006端口的訪問,當(dāng)然,轉(zhuǎn)發(fā)到本地某一端口不是限定的,可自由選擇。