美文网首页
Pytorch可视化工具visdom远程连接启动过程

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

作者: asl_1da7 | 来源:发表于2020-06-25 15:43 被阅读0次

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

    1. ssh远程服务器时绑定本地端口与远程服务器端口
    # ssh -L <local port>:localhost:<remote port> -p <port> username@remote_host
    ssh -L 6606:localhost:8097 -p <port> username@remote_host
    
    • 这里的6606是本地的端口
    • 8097是绑定的远程服务器的端口,选用8097的原因是visdom默认的端口是8097,可以修改。
    1. 在remote host开一个tmux窗口上启动visdom服务(这一步很关键),不启动visdom服务的话会报ConnectionError
    > visdom -p <remote port> --hostname <hostname>
    
    1. 模型代码中visdom的配置
    # server =  <hostname>
    # remote_port  = <remote port>
    visdom.Visdom(server=server, port = remote_port )
    
    • 这里的server启动visdom时指定的hostname
    • 这里的port是启动visdom时指定的remote port(上面的8097)
    1. 在local host的浏览器访问127.0.0.1:6606既可以查看模型的训练情况

    N.B

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

    相关文章

      网友评论

          本文标题:Pytorch可视化工具visdom远程连接启动过程

          本文链接:https://www.haomeiwen.com/subject/pomsfktx.html