在 Python 中尝试将 XGBoost 与 GPU 结合使用时出现错误“检查失败：gpu_predictor_:”

Question

我正在尝试在 Python 中使用 XGBoost，但出现错误

    xgboost.core.XGBoostError: [09:26:06] c:\users\administrator\workspace\xgboost-win64_release_1.0.0\src\gbm\gbtree.h:308: Check failed: gpu_predictor_:

我正在使用 Sklearn 并像这样构建模型

    model = XGBRegressor(learning_rate = 0.02,
                 n_estimators = 1000,
                    max_depth = 12,
                    tree_method='gpu_hist',
                    objective = 'reg:logistic',
                    nthread = -1,
                    silent = False)

它与 "hist" 一起作为 tree_method 在 cpu 上工作。

Answer 1

可能您的 GPU 不支持 XG Boost website 要求的 CUDA:

The GPU algorithms in XGBoost require a graphics card with compute capability 3.5 or higher, with CUDA toolkits 10.0 or later

至少，这是我的问题。如果要查看，可以使用this list.

遗憾的是，我找不到解决此问题的方法。

在 Python 中尝试将 XGBoost 与 GPU 结合使用时出现错误“检查失败：gpu_predictor_:”

Getting Error " Check failed: gpu_predictor_: " when trying to use XGBoost with GPU in Python

python

gpu

xgboost