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

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

python

Python GPU上的XGBoost正在杀死内核（在Ubuntu上）,python,gpu,xgboost,Python,Gpu,Xgboost,每次我使用以下代码在GPU上运行XGBoost时，内核都会死掉：参数={'tree_method'：'gpu_exact'，'max_depth'：100，'learning_rate'：0.1} dtrain=xgb.DMatrix（X_列，y_列） xgb.列车（参数、数据列车、evals=[（数据列车，“列车”））我得到： Kernel died - restarting 我认为主要的问题是max_depth=100，xgboost逐层构建树，因此深度为100的完整树将太大。例如，尝

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

参数={'tree_method'：'gpu_exact'，'max_depth'：100，'learning_rate'：0.1} dtrain=xgb.DMatrix（X_列，y_列） xgb.列车（参数、数据列车、evals=[（数据列车，“列车”））

我得到：

Kernel died - restarting

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

顺便说一句，如果您使用的是GPU渐变增强，我建议您尝试catboost，它是目前最快的GPU GBDT库。以下是英伟达博客博客：