无法从 keras.utils 导入 multi_gpu_model

Question

我在 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。

Answer 1

pip uninstall keras 
pip install keras==2.2.4

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

Answer 2

该功能于 10 月 11 日添加，目前还没有包含该功能的 Keras 版本。目前 Keras 的最后一个版本是 2017 年 8 月 25 日发布的 2.0.8。

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