Tensorflow 2.1.0 中 keras.utils.plot_model()出错的解决方法

 

小小和蓝胖 2020-02-15 21:31:36  370  收藏 2
展开
最近在学习Tensorflow 2，使用keras模型可视化的plot_model()绘制模型图的时候报错。上网查找解决方案，发现网上的解决方法没有切中核心，因此在此记录备查。

问题
希望绘制模型图：

from tensorflow import keras
keras.utils.plot_model(model, 'my_first_model.png')
1
2
报错：

Failed to import pydot. You must install pydot and graphviz for `pydotprint` to work.
1
解决方法
pip3 install pydot
sudo apt install graphviz
1
2
pydot是Python包，因此可以使用pip直接安装。
graphviz是贝尔实验室开源的做图软件。graphviz需要使用apt安装。如果使用pip3 install graphviz进行安装，虽然也能安装上，但是使用plot_model时仍然会报错。

附上结果图