ImportError: cannot import name ‘multi_gpu_model‘ from ‘keras.utils‘(Tensorflow 2.7.0)
·
1、问题
假设已经pip install keras(2.7.0版本)和tensorflow/tensorflow_gpu(2.7.0版本)了。
以keras为例:
(下载好之后,Pycharm下面Terminal的窗口会提示成功下载好的版本号。两者目前都是2.7.0版本。我是直接下载最新的,不用指定version)
我们同样可以在控制台打印出版本号:
import tensorflow
import keras
print(keras.__version__)
print(tensorflow.__version__)
问题出在下面的代码中,
from keras.utils import multi_gpu_model
这时会报错:
2、解决方法
通过查找Cannot import multi_gpu_model from keras.utils,我发现将上面的代码改为下面的语句,就不会报错了:
from tensorflow.python.keras.utils.multi_gpu_utils import multi_gpu_model
有帮助请给个支持,谢谢~
AtomGit 是由开放原子开源基金会联合 CSDN 等生态伙伴共同推出的新一代开源与人工智能协作平台。平台坚持“开放、中立、公益”的理念,把代码托管、模型共享、数据集托管、智能体开发体验和算力服务整合在一起,为开发者提供从开发、训练到部署的一站式体验。
更多推荐


所有评论(0)