小编Ehs*_*thi的帖子

如何在 XGBoost 中释放 GPU 上的所有内存?

这是我的代码:

clf = xgb.XGBClassifier(
  tree_method = 'gpu_hist',
  gpu_id = 0,
  n_gpus = 4,
  random_state = 55,
  n_jobs = -1
)
clf.set_params(**params)
clf.fit(X_train, y_train, **fit_params)
Run Code Online (Sandbox Code Playgroud)

我已阅读有关此问题和此git 问题的答案,但都不起作用。

我尝试用这种方式删除助推器:

clf._Booster.__del__()
gc.collect()
Run Code Online (Sandbox Code Playgroud)

它会删除助推器,但不会完全释放 GPU 内存。

我想它Dmatrix仍然在那里,但我不确定。

我怎样才能释放整个内存?

python gpu xgboost

8
推荐指数
1
解决办法
3887
查看次数

标签 统计

gpu ×1

python ×1

xgboost ×1