【Tensorflow】AttributeError: '_TfDeviceCaptureOp' object has no attribute '_set_device_from_string'

Question:

When using keras' multi_gpu_model

from keras.utils import multi_gpu_modelmodel = load_model()p_model = multi_gpu_model(model)

An error occurred:

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

The operating environment is:

tensorflow=1.15.0

keras=2.2.4

Resolve:

The reason for this error is The two versions do not match. Using keras=2.2.4 requires a matching tensorflow version, such as 1.13.1.If you want to use tensorflow1.15.0, you can use keras=2.3.1.

Version match reference: