未知的树更新程序 Growth_gpu_hist xgboost

Question

我已经用 训练了xgboost模型GPU('gpu_tree')。然后使用保存我的模型pickle。现在，我尝试在非 GPU 机器中加载该模型，但它抛出此错误。

raise XGBoostError(py_str(_LIB.XGBGetLastError()))
 xgboost.core.XGBoostError: [16:36:57] /tmp/pip-install-
mqijktew/xgboost/build/temp.linux-x86_64-3.6/xgboost/src/tree/
tree_updater.cc:20:
 Unknown tree updater grow_gpu_hist

任何帮助，将不胜感激。

Answer 1

前几天我遇到了同样的错误，问题是我使用 conda 安装了 xgboost，并且该软件包不包含 GPU 组件（看看 xgboost.dll，它只有几 MB）。然后我使用pip安装了xgboost（pip install xgboost），检查xgboost.dll的大小，它现在有300+ MB，并且这个错误“未知的树更新程序grow_gpu_hist”消失了。（xgboost版本为1.5.1）

不确定这是否有帮助，因为我注意到您的描述中的路径包含“pip-install”...