?

小小和藍胖 2020-02-15 21:31:36 ?370 ?收藏 2
展開
最近在學(xué)習Tensorflow 2，使用keras模型可視化的plot_model()繪制模型圖的時候報錯。上網(wǎng)查找解決方案，發(fā)現(xiàn)網(wǎng)上的解決方法沒有切中核心，因此在此記錄備查。

問題
希望繪制模型圖：

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時仍然會報錯。

附上結(jié)果圖

?

總結(jié)

以上是生活随笔為你收集整理的Tensorflow 2.1.0 中 keras.utils.plot_model()出错的解决方法的全部內(nèi)容，希望文章能夠幫你解決所遇到的問題。

如果覺得生活随笔網(wǎng)站內(nèi)容還不錯，歡迎將生活随笔推薦給好友。