这是我的代码:
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)
我尝试用这种方式删除助推器:
clf._Booster.__del__()
gc.collect()
Run Code Online (Sandbox Code Playgroud)
它会删除助推器,但不会完全释放 GPU 内存。
我想它Dmatrix仍然在那里,但我不确定。
我怎样才能释放整个内存?