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

华为开发者空间,是为全球开发者打造的专属开发空间,汇聚了华为优质开发资源及工具,致力于让每一位开发者拥有一台云主机,基于华为根生态开发、创新。

更多推荐