PyCharm TensorFlow Plot_model 无法工作

Posted 2023-03-28

【中文标题】PyCharm TensorFlow Plot_model 无法工作

【英文标题】：PyCharm Tensorflow Plot_model cannot work

【发布时间】：2021-09-02 08:59:00

【问题描述】：

我在 PyCharm 中使用 TensorFlow 显示绘图模型时遇到问题。

我从 PyCharm 的 Setting 部分的 Python Interpreter 安装了 pydot 和 graphviz。

接下来，我运行代码并得到一个错误，定义如下。

我该如何解决？

这是我的代码 sn-p，如下所示。

from tensorflow.keras.utils import plot_model

# Plot Model
plot_model(model, show_shapes=True)

这是我的错误，如下所示。

ImportError: ('You must install pydot (`pip install pydot`) and install graphviz (see instructions at https://graphviz.gitlab.io/download/) ', 'for plot_model/model_to_dot to work.

【参考方案1】：

这是我的解决方案：

1) 从https://graphviz.org/download/下载graphviz

2 ) 定义 路径 以从 Tensorflow 识别 plot_model

import os
os.environ["PATH"] += os.pathsep + 'C:/Program Files/Graphviz/bin/'

3 ) 定义 plot_model 如下所示。

# Plot Model
plot_model(model, "model.png" , show_shapes=True, show_layer_names=True)