XGBoost를 GPU로

conda-forge의 XGBoost는 아직 GPU를 지원하지 않는다고 한다.
구글링해서 XGBoost를 따로 설치하고 설정해도 돌아가지 않았는데,
nvidia가 제공하는 XGBoost를 설치하면 이용할 수 있다고 한다.

conda remove xgboost
conda install -c nvidia -c rapidsai py-xgboost

# parameter (자세한 설명은 다음에. 일단 기록)
'gpu_id': 0
'tree_method': 'gpu_hist'
'predictor': 'gpu_predictor'

• parameter 설명

고생해서 찾았는데 한참에 걸리던 녀석이 금방 끝나는 걸 보니 뿌듯하다.

• 참고 : https://github.com/dmlc/xgboost/issues/5447