pytorch安裝visdom及報錯的處理辦法

原創

小研一枚

2020-02-20 14:53

兩條命令：

安裝visdom：pip install visdom
在cmd中：輸入： python -m visdom.server，開啓web服務。
我在運行第一步沒有問題，主要是在第二步開啓web服務時，總是報錯大概是 it take a little while to download scripts之類的，等了一會後報錯無法找到路徑。

解決方法：

通過繼續輸入pip install visdom 找到visdom存放路徑，然後打開裏面的server.py文件夾，修改最下面的download_scripts_and_run()函數：註釋掉下載部分就行了。

再運行第二步即可安裝成功。如下：

然後在瀏覽器輸入：http://localhost:8097 ，即跳出界面。

例子：

新建以下python，並運行。然後在瀏覽器輸入：http://localhost:8097 ，左上角由mian改爲test1，刷新網頁既有如下輸出。

import torch
import visdom

vis = visdom.Visdom(env='test1')
x = torch.arange(1, 30, 0.01)
y = torch.sin(x)
vis.line(X=x, Y=y, win='sinx', opts={'title': 'y=sin(x)'})

# append 追加數據
for ii in range(0, 10):
    # y = x
    x = torch.Tensor([ii])
    y = x
    vis.line(X=x, Y=y, win='polynomial', update='append' if ii > 0 else None)

# updateTrace 新增一條線
x = torch.arange(0, 9, 0.1)
y = (x ** 2) / 9
vis.line(X=x, Y=y, win='polynomial', name='this is a new Trace')

# 可視化一個隨機的黑白圖片
# vis.image(torch.randn(64, 64).numpy())

# 隨機可視化一張彩色圖片
vis.image(torch.randn(3, 64, 64).numpy(), win='random2')

# 可視化36張隨機的彩色圖片，每一行6張
vis.images(torch.randn(36, 3, 64, 64).numpy(), nrow=6, win='random3', opts={'title': 'random_imgs'})

vis.text(u'''<h1>Hello Visdom</h1><br>Visdom是Facebook專門爲<b>PyTorch</b>開發的一個可視化工具，
         在內部使用了很久，在2017年3月份開源了它。
         Visdom十分輕量級，但是卻有十分強大的功能，支持幾乎所有的科學運算可視化任務''',
         win='visdom',
         opts={'title': u'visdom簡介'}
         )