无法从keras.utils导入multi_gpu_model

时间:2017-10-19 22:53:20

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

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

我无法执行:

Add-Migration Init
Update-Database

有没有人成功使用multi_gpu_model,如文档的FAQ部分所述?

我有一台带有4块GeForce GTX 1080 Ti卡的4 GPU机器,想要全部使用它们。

这是我得到的错误:

from kears.utils import multi_gpu_model 

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

5 个答案:

答案 0 :(得分:6)

pip uninstall keras 
pip install keras==2.2.4

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

答案 1 :(得分:3)

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

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

答案 2 :(得分:2)

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

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

答案 3 :(得分:2)

除了Matias评论,你也可以

https://github.com/fchollet/keras/blob/master/keras/utils/training_utils.py

复制multi_gpu_model代码

答案 4 :(得分:0)

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

TAB_1

我由此得以解决了这个问题。