CODEWITHSUNDEEP
pythongpuxgboost
Sam Karkach
Sam Karkach

Reputation: 21

XGBoost on GPU is killing the kernel

The kernel dies every time I run XGBoost on GPU with this code:

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")])

I get:

Kernel died - restarting

I am in Ubuntu.

Upvotes: 2

Views: 1088

Answers (1)

Anna Veronika Dorogush
Anna Veronika Dorogush

Reputation: 1233

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.

Upvotes: 2

Related Questions