GPU上的XGBoost正在杀死内核（在Ubuntu上）

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

Answer 1

I think the main problem is max_depth=100, xgboost builds trees layer by layer so a full tree of depth 100 will be too large. Try using values up to 12 for example.