是否可以在GPU上运行catboost预测?

时间:2018-11-30 20:45:11

标签: machine-learning catboost

我有一个用例,需要对非常大的数据集进行预测。因此,与训练相比,预测花费了将近10倍的时间。我想知道我是否可以使用GPU进行预测,或者它是否受CPU限制。我使用的是EC2 P3.16x大(8x Tesla V100,64个VCPU)

1 个答案:

答案 0 :(得分:0)

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

对于某些应用,Catboost可能会受益于GPU预测器(例如,许多决策树),但我们没有资源在不久的将来实现。