tensorflow-gpu 2.4.x保存模型图‘Failed to import pydot. You must `pip install pydot` and install graphviz

为了更加方便地查看tensorflow的模型，准备尝试用tensorflow的plot_model画出模型图，但是，尝试运行保存模型的代码：

from keras.utils.vis_utils import plot_model

plot_model(auto, to_file='./model.png', show_shapes=True, show_layer_names=True, rankdir='TB')
img = plt.imread('model.png')
plt.figure(figsize=(10, 10))
plt.imshow(img)
plt.axis('off')
plt.show()

得到一个错误：

('Failed to import pydot. You must `pip install pydot` and install graphviz (https://graphviz.gitlab.io/download/), ', 'for `pydotprint` to work.')

尝试了下安装pydot，直接在命令行运行

pip install pydot

再次运行绘制模型的代码，发现还是提示同样的错误，看到这篇博客：
win10中TensorFlow2.1实现模型可视化（plot_model函数）解决出错 ImportError: Failed to import pydot.
这篇博客提到在命令行中输入

dot -version

正常出现dot才表示成功，尝试在命令行中输入dot -version，结果提示如下：

Command 'dot' not found, but can be installed with:

sudo apt install graphviz

原来如此，还需要安装graphviz，那就只在命令行输入

sudo apt install graphviz

再次运行代码，模型图终于出来了