model = Model(inputs=[v_i, v_j], outputs=output_list) model = multi_gpu_model(model,4) model.compile(....)

主要就是第二句话中的

multi_gpu_model函数，会把数据和模型分到多个gpu上执行

有个坑，就是整个程序导入keras时要么全部from keras import ...,要么全部 from tensorflow.python.keras import ...，不能tensorflow里的keras不能和原生keras混用，
要不就会报Node' object has no attribute 'output_masks的错误
扫码关注我们
微信号：SRE实战
拒绝背锅 运筹帷幄

SRE实战 互联网时代守护先锋，助力企业售后服务体系运筹帷幄！一键直达领取阿里云限量特价优惠。