ImportError:无法导入`pydot`。请安装`pydot`。例如`pip install pydot`

时间:2019-03-29 08:40:17

标签: python keras ipython pydot

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

  

ImportError:无法导入pydot。请安装pydot。例如>与pip install pydot

使用的代码:

例如模型= 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,然后再次安装了它。但是错误仍然存​​在。

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

1 个答案:

答案 0 :(得分:2)

如果您在jupyter笔记本中遇到此错误,则有可能您使用的是与系统默认设置不同的python解释器。

在笔记本中尝试以下操作:

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

此代码段会将当前python解释器的路径传递到终端,并以pip作为模块运行以安装库。

检查此链接以获取更多信息

如果您正在使用其他解释器,则首先发现您正在使用哪个解释器: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

但是我更喜欢从终端安装库并保持代码干净。