【问题标题】:Cannot import multi_gpu_model from keras.utils无法从 keras.utils 导入 multi_gpu_model
【发布时间】:2018-04-01 02:09:57
【问题描述】:

我在 ubuntu 16.04 上有 tensorflow-gpu 1.2.1 和 keras。

我无法执行:

from kears.utils import multi_gpu_model 

有没有人在使用 multi_gpu_model 时取得了成功,如文档的常见问题部分所述?

我有一台带有 4 个 GeForce GTX 1080 Ti 卡的 4 GPU 机器,我想使用所有这些卡。

这是我得到的错误:

import keras.utils.multi_gpu_model

---------------------------------------------------------------------------
ModuleNotFoundError                       Traceback (most recent call last)
<ipython-input-7-0174878249b1> in <module>()
----> 1 import keras.utils.multi_gpu_model
      2 

ModuleNotFoundError: No module named 'keras.utils.multi_gpu_model'

我可以成功导入keras和keras.utils。

【问题讨论】:

  • 如果您提供了从该行得到的错误,这将有所帮助。
  • ------------------------------------------ --------------------------------- ImportError Traceback(最近一次调用最后一次)() ----> 1 from keras.utils import multi_gpu_model 2 ImportError: cannot import name 'multi_gpu_model'
  • 请使用此新信息编辑问题(问题下方有一个“编辑”链接),而不是在 cmets 中提供。它将帮助能够回答此问题的人更快地找到信息。
  • 您使用from kears.utils import multi_gpu_model 还是import keras.utils.multi_gpu_modelmulti_gpu_model是一个函数,所以不能使用后者。
  • 另外请检查您的 Keras 版本。您需要从 GitHub 安装最新版本才能使用此功能。即使是最新发布的 Keras 2.0.8 也不够新。

标签: python neural-network deep-learning keras gpu


【解决方案1】:

这个功能是在 10 月 11 日添加的,目前还没有包含它的 Keras 版本。目前最新的 Keras 版本是 2017 年 8 月 25 日发布的 2.0.8。

您唯一的选择是等到发布,或者使用 Keras 直接来自 git master 的版本。

【讨论】:

  • 谢谢马蒂亚斯。我从 git 安装了最新版本,该功能运行良好。
【解决方案2】:

要关闭此问题:我从 github 安装了最新版本的 Keras,并让 multi_gpu_model 模块按预期工作。

pip install git+git://github.com/fchollet/keras.git --upgrade

【讨论】:

    【解决方案3】:

    您可以像这样重新安装 tensorflow-estimator 和 keras:

    pip uninstall tensorflow-estimator
    pip install tensorflow-estimator
    pip uninstall keras
    pip install keras
    

    我能够通过这个解决问题。

    【讨论】:

      【解决方案4】:
      pip uninstall keras 
      pip install keras==2.2.4
      

      这对我有帮助... 在 keras hrnetv2 上工作

      【讨论】:

        【解决方案5】:

        对于 Tensorflow 2.7.0 这有效:

        from tensorflow.python.keras.utils.multi_gpu_utils import multi_gpu_model
        

        【讨论】:

          猜你喜欢
          • 2021-07-21
          • 1970-01-01
          • 1970-01-01
          • 2021-08-05
          • 1970-01-01
          • 2021-07-05
          • 1970-01-01
          • 2019-07-12
          • 1970-01-01
          相关资源
          最近更新 更多