1.正确使用的流程:
我的环境是tensorflow 2.6.0,python3.9.18。
-
安装对应的库
pip install graphviz
pip install pydotplus -
安装文件graphviz.smi,我安装的是8.1.0版本。
下载地址:graphviz.smi
安装的时候记得勾选环境变量选项。
- 修改vis_utils.py ,将pydot 的都替换成pydotplus。原因是pydot已经停止开发了,不兼容了。
如何找到vis_utils.py?
在pycharm中把鼠标放在plot_model函数的位置,然后Ctrl+单击该函数即可进入。
直接一键全部替换即可: