Pytorch可视化工具visdom远程连接启动过程

visdom是由Facebook开发的PyTorch训练过程可视化工具，类似于tensorboard。本文主要记录在远程服务器上跑PyTorch模型时如何在本地访问visdom的结果。

1. ssh远程服务器时绑定本地端口与远程服务器端口

# ssh -L :localhost: -p  username@remote_host
ssh -L 6606:localhost:8097 -p  username@remote_host

• 这里的6606是本地的端口
• 8097是绑定的远程服务器的端口，选用8097的原因是visdom默认的端口是8097，可以修改。

2. 在remote host开一个tmux窗口上启动visdom服务（这一步很关键），不启动visdom服务的话会报ConnectionError

> visdom -p  --hostname 

3. 模型代码中visdom的配置

# server =  
# remote_port  = 
visdom.Visdom(server=server, port = remote_port )

• 这里的server启动visdom时指定的hostname
• 这里的port是启动visdom时指定的remote port(上面的8097)

4. 在local host的浏览器访问127.0.0.1:6606既可以查看模型的训练情况

N.B

• 一定要先启动visdom服务
• 在本地可以打开127.0.0.1:6606证明已经连接成功了，没有内容可能是第一个iter还没跑完