Python Xgboost（GPU）在预测时崩溃

我正在使用Python中的XGBoost GPU版本，每当我尝试运行时它就会崩溃。它适用于较小的数据集，但对于我目前的问题，它不起作用

train_final.shape, test_final.shape
((631761, 174), (421175, 174))


params = {
          'objective': 'multi:softmax', 
          'eval_metric': 'mlogloss',
          'eta': 0.1,
          'max_depth': 6, 
          'nthread': 4,
          'alpha':0,
          'num_class': 5,
          'random_state': 42, 
          'tree_method': 'gpu_hist',
          'silent': True
             }

GPU统计：GTX 1070，6 GB 内存：32 GB

有人能帮我理解为什么会这样吗？

---

谢谢分享

# training
bst = xgb.train(param, dtrain, num_round)

#save model
joblib.dump(bst, 'xgb_model.dat')
bst.__del__()

#load saved model
bst = joblib.load('xgb_model.dat')
preds = bst.predict(dtest)