文章来源 | 恒源云社区(专注人工智能/深度学习云GPU服务器训练平台,十一月宠粉活动ing,官方体验网址:https://gpushare.com/ )
原文地址 | https://gpushare.com/docs/bes...
Visdom
众所周知,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!')