要想使用
GPU
训练,需要指定tree_method
参数为下列的值:'gpu_exact'
: 标准的xgboost
算法。它会对每个分裂点进行精确的搜索。相对于'gpu_hist'
,它的训练速度更慢,占用更多内存
当 为
'gpu_exact','gpu_hist'
时,模型的predict
默认采用GPU
加速。'cpu_predictor'
: 使用CPU
来执行模型预测'gpu_predictor'
: 使用GPU
来执行模型预测
多
GPU
不一定比单个 更快,因为PCI
总线的带宽限制,数据传输速度可能成为瓶颈。