【工具篇5】如何安装、使用可视化利器 Visdom？

[Alice_恒源云]

众所周知，Visdom 是 Facebook 开发的一款 PyTorch 数据可视化工具～

那如何在我们平台安装使用它呢？本篇教程将简单说说～

【准备】

在安装前，提醒大家一下，平台实例并不提供公网 IP 地址，服务都是通过端口映射到公网接入点提供链接的～

因此，使用 Visdom 时，需要停止 TensorBoard 服务，并且将 Visdom 端口配置成和 TensorBoard 相同的 6006 端口上，监听地址需要使用 0.0.0.0。

【安装】

完成以上步骤后，就可以正式安装啦～

在终端输入以下命令即可：

pip install visdom

#替换下载静态文件的地址，防止卡 download
sed -i "s/https:\/\/\(cdnjs.cloudflare.com.*\)/http:\/\/\1/" "$(pip show visdom | grep Location | awk '{print $2}')/visdom/server.py"

#停止 TensorBoard 服务
supervisorctl stop tensorboard

#如果需要禁止 TensorBoard 随实例启动，使用如下命令配置
grep -E "autostart" /etc/supervisor/conf.d/tensorboard.conf || echo "autostart = false" >>/etc/supervisor/conf.d/tensorboard.conf
supervisorctl update

#启动 Visdom 服务
visdom --hostname 0.0.0.0 -port 6006

安装完成后，点击页面 TensorBoard 的链接，实际进入的就是 Visdom 啦～

（官方透露一个小技巧，理论上，其他 Web 服务也可以用这种方法连接，将 Visdom 换成其他工具，也行得通～

【使用】

要注意喔，Visdom 服务本地启动在了 6006 端口，在 Python 客户端初始化时，需要添加参数 port=6006，如下：

import visdom
import numpy as np

vis = visdom.Visdom(port=6006)
vis.text('Hello, world!')

[Alice_恒源云]

本篇教程参考了技术小哥哥编写的文档，如有任何疑问，请在本帖下留言哈

再附上一个文档原址：https://gpushare.com/docs/best_practices/visdom/