假设已经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
通过查找Cannot import multi_gpu_model from keras.utils,我发现将上面的代码改为下面的语句,就不会报错了:
from tensorflow.python.keras.utils.multi_gpu_utils import multi_gpu_model
有帮助请给个支持,谢谢~