Tensorflow2.X版本使用keras.utils.plot_model()所需的安装

1、所需要的pip包

安装pydot、pydot_ng、graphviz

pip install -i https://pypi.tuna.tsinghua.edu.cn/simple pydot
pip install -i https://pypi.tuna.tsinghua.edu.cn/simple pydot_ng
pip install -i https://pypi.tuna.tsinghua.edu.cn/simple graphviz

2、手动安装graphviz

下载地址.
将安装好后的graphviz的bin文件夹添加到环境变量中

3、使用keras.utils.plot_model()进行模型绘制

import tensorflow as tf
from tensorflow import keras

import pydot
keras.utils.plot_model(model)