XGBoost на графическом процессоре убивает ядро ⇐ Python

[Anonymous]

Ядро умирает каждый раз, когда я запускаю 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.

Подробнее здесь: https://stackoverflow.com/questions/533 ... the-kernel