标签: python-3.x tensorflow keras multi-gpu
我有很多张量运算(matmul,转置等)。我想在大型数据集上运行。 由于它们仍然是矩阵运算,并且由于我使用Keras生成器来加载数据批,因此使用GPU进行计算将很有意义。
现在,我已经搜索了一段时间,但似乎找不到使用Keras在标准Model对象界面之外使用生成器进行并行GPU操作的正确方法。
Model
有人知道怎么做吗?谢谢!