最近在学习
Tensorflow 2
,使用keras
模型可视化的plot_model()
绘制模型图的时候报错。上网查找解决方案,发现网上的解决方法没有切中核心,因此在此记录备查。
问题
希望绘制模型图:
from tensorflow import keras
keras.utils.plot_model(model, 'my_first_model.png')
报错:
Failed to import pydot. You must install pydot and graphviz for `pydotprint` to work.
解决方法
pip3 install pydot
sudo apt install graphviz
pydot
是Python包,因此可以使用pip
直接安装。
graphviz
是贝尔实验室开源的做图软件。graphviz
需要使用apt
安装。如果使用pip3 install graphviz
进行安装,虽然也能安装上,但是使用plot_model
时仍然会报错。