问题:
使用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。
版本匹配参考: