XGBoost GPU 로 가속하기

출처: https://xgboost.readthedocs.io/en/stable/gpu/index.html

XGBoost GPU Support — xgboost 1.5.2 documentation

 

CUDA 10.0 이상의 버전이 설치되어 있으면 GPU로 XGBoost를 가속할 수 있다.

 

방법은 다음과 같다.

모델에 다음과 같이 tree_method와 gpu_id parameter를 추가해 주면 된다.

XGBRegressor(tree_method='gpu_hist', gpu_id=0)

 

gpu_id는 사용할 gpu의 번호로, nvidia-smi를 이용해 확인할 수 있다.