machine-learning - 是否可以在 GPU 上运行 catboost predict？

Question

我有一个需要在非常大的数据集上进行预测的用例。因此，与训练相比，预测所需的时间几乎是 10 倍。我想知道我是否可以使用 GPU 进行预测，或者它是否受 CPU 限制。我正在使用 EC2 P3.16x large（8x Tesla V100、64 VCPU）

Answer 1

我们还没有 GPU 预测器，只有在测试集上的学习、度量计算和预测计算。

对于某些应用程序，Catboost 可以从 GPU 预测器（例如许多决策树）中受益，但我们在不久的将来没有资源可以实现。