Keras使用多個GPU並行

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的錯誤


https://blog.csdn.net/sinat_26917383/article/details/75633754上邊博客裏邊動態分配內存時，若是使用的是tensorflow裏的keras那麼不用set_session,若是遠程keras那麼須要K.set_session(sess)