【Tensorflow】AttributeError: ‘

使用keras的multi_gpu_model时

from keras.utils import multi_gpu_model

model = load_model()
p_model = multi_gpu_model(model)

出现错误：

AttributeError: '_TfDeviceCaptureOp' object has no attribute '_set_device_from_string'

运行环境是：

tensorflow=1.15.0

keras=2.2.4

出现该错误的原因是两者版本不匹配，使用keras=2.2.4需要匹配的tensorflow版本，例如1.13.1。如果想用tensorflow1.15.0则可以使用keras=2.3.1。

版本匹配参考：