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

  有帮助请给个支持,谢谢~

Logo

AtomGit 是由开放原子开源基金会联合 CSDN 等生态伙伴共同推出的新一代开源与人工智能协作平台。平台坚持“开放、中立、公益”的理念,把代码托管、模型共享、数据集托管、智能体开发体验和算力服务整合在一起,为开发者提供从开发、训练到部署的一站式体验。

更多推荐