개요

LightGBM 사용시 GPU 가속하기에 이어 XGBoost GPU 가속 방법을 정리해둔다.

XGBoost GPU 가속버전 빌드

$ git clone --recursive https://github.com/dmlc/xgboost
$ cd xgboost
$ mkdir build
$ cd build
$ cmake .. -DUSE_CUDA=ON
$ make -j4

파이썬 패키지 설치

먼저 본인이 원하는 virtualenv로 진입

$ cd python-package; python setup.py install

.bashrc에 아래의 내용을 추가한다.

export PYTHONPATH=$PYTHONPATH:/xgboost경로/python-package

파라미터 설정

tree_method 파라미터를 gpu_exactgpu_hist로 설정한다.

예를 들면 코드는 아래와 같다.

model = xgb.XGBRegressor(n_estimators=400, learning_rate=0.05, gamma=0, subsample=0.75, colsample_bytree=1,
                            max_depth=7, tree_method='gpu_exact')

두 파라미터의 차이 및 다양한 파라미터 설정은 https://xgboost.readthedocs.io/en/latest/gpu/를 참고한다.