ImportError:未能导入“`pydot`”。请安装“`pydot`”。例如，使用“`pip install pydot`”

Question

我尝试使用Ipython和keras.utis.vis_utils可视化一个Keras模型。但我收到以下错误。

ImportError:导入pydot失败。请安装pydot。例如，>使用pip install pydot。

所用代码：

G模型= ggv.h5

from IPython.display import SVG
from keras.utils.vis_utils import model_to_dot
SVG(model_to_dot(model).create(prog='dot', format='svg'))

回溯

Traceback (most recent call last):
  File "C:\Users\VW3ZTWS\PycharmProjects\Data_Collection_and_learnings\venv\lib\site-packages\IPython\core\interactiveshell.py", line 2869, in run_code
    exec(code_obj, self.user_global_ns, self.user_ns)
  File "<ipython-input-49-b1aadf48e3ac>", line 5, in <module>
    SVG(model_to_dot(model).create(prog='dot', format='svg'))
  File "C:\Users\VW3ZTWS\PycharmProjects\Data_Collection_and_learnings\venv\lib\site-packages\keras\utils\vis_utils.py", line 55, in model_to_dot
    _check_pydot()
  File "C:\Users\VW3ZTWS\PycharmProjects\Data_Collection_and_learnings\venv\lib\site-packages\keras\utils\vis_utils.py", line 20, in _check_pydot
    'Failed to import `pydot`. '
ImportError: Failed to import `pydot`. Please install `pydot`. For example with `pip install pydot`.

我试图卸载pydot，graphviz，然后再次安装它。但错误依然存在。

在这种情况下，什么才是可视化模型的最佳解决方案？

Answer 1

如果您在jupyter笔记本中获得了该错误，则有可能使用与系统默认值不同的python解释器。

在笔记本里试试这个：

import sys
%$sys.executable -m pip install pydot

这个片段将把当前python解释器的路径传递给终端，并作为一个模块运行pip来安装库。

查看此链接以获取更多信息。

• https://docs.python.org/3/installing/index.html
• https://ipython.readthedocs.io/en/stable/interactive/magics.html

如果您使用的是另一个解释器，那么首先发现您使用的是哪个解释器：print(sys.executable)

然后使用该信息在终端中运行：path_to_the_interpeter -m pip install pydot

另一种方法是在代码中执行此操作：

import subprocess
import sys

def install(library):
    subprocess.call([sys.executable, "-m", "pip", "install", library])

try:
    import pydot
except ImportError:
    install('pydot')
    import pydot

但我更喜欢从终端安装库，并保持代码整洁。