python - 无法从 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 block 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 上工作