分类：gpu并行

2020
10-07

keras 多gpu并行运行案例

一、多张gpu的卡上使用keras有多张gpu卡时，推荐使用tensorflow作为后端。使用多张gpu运行model，可以分为两种情况，一是数据并行，二是设备并行。二、数据并行数据并行将目标模型在多个设备上各复制一份，并使用每个设备上的复制品处理整个数据集的不同部分数据。利用multi_gpu_model实现keras.utils.multi_gpu_model(model,gpus=None,cpu_merge=True,cpu_relocation=False)具体来说，该功能实现了单机多GPU数据并行性。它... 继续阅读 >