远程GPU服务器训练时Tensorboard的本地可视化方法

做AI&ML课大作业时遇到的问题,解决后记录一下。

解决问题:

由于tensorflow程序在远程GPU服务器运行,而tensorboard启动后访问地址为:0.0.0.0:6006 / localhost:6006 ,按理说用VPN在内网应该可以访问xx.xx.xx.xx:6006 打开tensorboard面板可视化训练过程的,但因不明原因始终ping不通,疑似tensorboard特性?,visdom无此问题,直接导致无法在本机用浏览器访问。

解决方案

利用Xshell隧道
首先 隧道 在Xshell > 文件 > 属性 > 连接 > SSH > 隧道 
点添加 ,保持源主机和目标主机 都是 localhost不变
直接将侦听端口 和 目标端口 都设置成6006(你设定的tensorboard的访问端口)
file

运行方法

在服务器端运行nohup tensorboard --logdir=='runs'(计算图文件夹) --port=6006 --host='localhost' &
然后本机浏览器访问 localhost:6006 or 127.0.0.1:6006
file