July · 2021年11月19日

恒源云(GPUSHARE)_云GPU服务器如何使用 Visdom?

文章来源 | 恒源云社区(专注人工智能/深度学习云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!')

image

推荐阅读
目录
极术微信服务号
关注极术微信号
实时接收点赞提醒和评论通知
安谋科技学堂公众号
关注安谋科技学堂
实时获取安谋科技及 Arm 教学资源
安谋科技招聘公众号
关注安谋科技招聘
实时获取安谋科技中国职位信息