无法从 keras.utils 导入 multi_gpu_model

Question

I have tensorflow-gpu 1.2.1 and keras on ubuntu 16.04.

I am not able to perform:

from kears.utils import multi_gpu_model 

Has anyone had success with multi_gpu_model as described in their documentation's FAQ section?

I have a 4 GPU machine with 4 GeForce GTX 1080 Ti cards and want to use all of them.

Here's the error I get:

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'

I can import keras and keras.utils successfully.

Answer 1

pip uninstall keras 
pip install keras==2.2.4

this helped me ... working on keras hrnetv2

Answer 2

This function was added on October 11, and there have been no Keras releases that include it. Currently the last Keras release is 2.0.8 released on Aug 25, 2017.

Your only options are to wait until a release is made, or to use a Keras version directly from git master.

Answer 3

Except Matias comment, you could also

copy the multi_gpu_model code from https://github.com/fchollet/keras/blob/master/keras/utils/training_utils.py

Answer 4

要解决此问题：我从 github 安装了最新版本的 Keras，并让 multi_gpu_model 模块按预期工作。

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

Answer 5

对于 Tensorflow 2.7.0，这有效：

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

Answer 6

You could reinstall tensorflow-estimator and keras like this:

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

I was able to resolve the issue by this.