安装与启动环境

pip install visdom

启动服务

python -m visdom.server

得到地址

浏览器打开：

代码执行测试

找代码执行一下：

import  visdom
import visdom

# 新建一个连接客户端
# 指定env = 'test1'，默认是'main',注意在浏览器界面做环境的切换
vis = visdom.Visdom(env='test1')
# 绘制正弦函数
x = torch.arange(1, 100, 0.01)
y = torch.sin(x)
vis.line(X=x,Y=y, win='sinx',opts={'title':'y=sin(x)'})
# 绘制36张图片随机的彩色图片
vis.images(torch.randn(36,3,64,64).numpy(),nrow=6, win='imgs',opts={'title':'imgs'})

切换环境查看：

# 绘制36张图片随机的彩色图片
vis.images(torch.randn(36,3,64,64).numpy(),nrow=6, win='imgs',opts={'title':'imgs'})

torchvision也有相同的功能

import torch
from torchvision import transforms
to_pil = transforms.ToPILImage()
to_pil(torch.randn(3, 64, 64))