Keras在没有Model对象的情况下使用multi-gpu(不适用于训练)

时间:2019-06-18 17:59:45

标签: python-3.x tensorflow keras multi-gpu

我有很多张量运算(matmul,转置等)。我想在大型数据集上运行。 由于它们仍然是矩阵运算,并且由于我使用Keras生成器来加载数据批,因此使用GPU进行计算将很有意义。

现在,我已经搜索了一段时间,但似乎找不到使用Keras在标准Model对象界面之外使用生成器进行并行GPU操作的正确方法。

有人知道怎么做吗?谢谢!

0 个答案:

没有答案