目錄

1. 安裝 graphviz 和 torchviz

2.安裝 graphviz.exe

3.實例測試

4.如果你的電腦還是無法畫圖，并且出現(xiàn)了下面的報錯：

5.參考文章：

1. 安裝 graphviz 和 torchviz

首先打開?Anaconda?prompt 進入自己的 pytorch 環(huán)境(圖中 pt 是我自己的 pytorch 環(huán)境)，運行如下代碼安裝依賴包。

pip install graphviz torchviz

2.安裝 graphviz.exe

下載地址：Index of /Packages/stable/windows/10/cmake/Release/x64

下面選擇第二個，選擇將graphviz添加到環(huán)境變量。

只需要安裝到一個自己記得住的地方，我安裝的地方是 C:\Program Files\Graphviz 2.44.1

?

檢查是否已經(jīng)將將graphviz添加到環(huán)境變量中。

我們按照依次順序進入到編輯環(huán)境變量中：win左下角搜索 -> 查看高級系統(tǒng)設(shè)置 -> 點擊環(huán)境變量 ->點擊用戶的path。

在這里添加變量名和變量值即可，

變量名為：graphviz

變量值為：graphviz所在的路徑+bin；那么我的變量值就為：C:\Program Files\Graphviz 2.44.1\bin

3.實例測試

編寫一個實現(xiàn)vgg16的結(jié)構(gòu)的畫圖。

import torch
from torchviz import make_dot
from torchvision.models import vgg16  # 以 vgg16 為例x = torch.randn(4, 3, 32, 32)  # 隨機生成一個張量
model = vgg16()  # 實例化 vgg16，網(wǎng)絡可以改成自己的網(wǎng)絡
out = model(x)   # 將 x 輸入網(wǎng)絡
g = make_dot(out)  # 實例化 make_dot
g.view()  # 直接在當前路徑下保存 pdf 并打開
# g.render(filename='netStructure/myNetModel', view=False, format='pdf')  # 保存 pdf 到指定路徑不打開

執(zhí)行完成之后就可以看到畫出的結(jié)構(gòu)。?

4.如果你的電腦還是無法畫圖，并且出現(xiàn)了下面的報錯：

CalledProcessError: Command ‘[‘dot‘, ‘-Tpdf‘, ‘-O‘, ‘測試圖片.gv‘]‘ returned non-zero exit status 1. [st

那么我們這個時候就可以按照下面步驟完成畫圖：

1. 打開cmd（使用管理員身份，要不然沒有權(quán)限）
2. 運行：dot -v
3. 出現(xiàn)問題：There is no layout engine support for “dot”…
4. 運行：dot -c
5. 沒有提示（成功）
6. 再次運行dot -v
7. 結(jié)果就會畫圖成功

5.參考文章：

https://blog.csdn.net/kelanj/article/details/108394313pytorch 網(wǎng)絡可視化(二)：graphviz + torchviz_聽 風、的博客-CSDN博客