版本:Win10
Python:3.7
PyTorch 1.7.1


前言

在用PyTorch可视化.dot文件时,遇到了以下这种情况:
bug


问题解决

  起初以为是未安装graphviz包,pip一下发现依旧报错,后来查看资料才发现需要graphviz是一个程序,需要安装。

安装步骤:

  1. 安装graphviz

  2. 配置环境变量

  3. 重启

  4. 安装相关包

安装步骤

1.下载Graphviz

官网安装:
官网下载安装包:https://graphviz.org/download/

选择相应的系统:
g

根据本机选择相应版本:我选的是windows版64位,安装版(exe格式)
g
下载的安装包:
g


2.安装Graphviz

打开安装包,点击 下一步
g
点击 我接受(I)
g
  为了避免之后忘了添加环境变量,我在这里就选了 直接自动添加环境变量,并允许所有用户都可以使用,之后点击 下一步(N) :(读者可以按自己需求选择,包括后面那个生成桌面图标)
g
选择安装目录,我是默认,之后点击 下一步(N)
g
这里默认即可,点击 安装(I)
g
等待安装完成:
g
点击 完成(F)
g


3.配置环境变量

如果安装时没有选择自动配置环境变量,这里需要配置以下,因为不配置的话又会发生最开始时的错误!
g
最后输入 dot -version 验证是否成功
g


4.重启(我感觉很是需要!!!)

5.安装相关的包

安装graphviz
g
下面是我的需要, 安装了其他的包。
g
g

6.验证代码(成功)

g

总结

以上是问题的解决办法,希望能帮助到大家!

Logo

一站式 AI 云服务平台

更多推荐