GPU 上的 XGBoost 正在杀死内核

Question

每次我使用以下代码在 GPU 上运行 XGBoost 时，内核都会死掉：

params = {'tree_method': 'gpu_exact', 'max_depth': 100, 'learning_rate': 0.1}
dtrain = xgb.DMatrix(X_train,y_train )
xgb.train(params, dtrain, evals=[(dtrain, "train")])

我得到：

Kernel died - restarting

我在Ubuntu中。

Answer 1

我认为主要问题是max_深度=100，xgboost逐层构建树，所以深度100的完整树会太大。例如，尝试使用最大 12 的值。