我在 ubuntu 16.04 上有 tensorflow-gpu 1.2.1 和 keras。
我无法执行:
from kears.utils import multi_gpu_model
是否有人按照其文档的常见问题解答部分中的描述成功使用了 multi_gpu_model?
我有一个 4 GPU 机器和 4 个 GeForce GTX 1080 Ti 卡,我想使用所有这些。
这是我得到的错误:
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。
原文由 Sharanya Arcot Desai 发布,翻译遵循 CC BY-SA 4.0 许可协议
这帮助了我……在keras hrnetv2上工作