그래디언트 부스팅 (gradient goosting) 을 사용하기 쉽게 구현해둔 패키지는 LightGBM 이나 XGBoost가 있다.
XGBoost 의 경우 콘다 환경에서 간단하게 pip install xgboost 로 설치해도 GPU 사용 방법이 매우 단순하다.
예를 들어 아래와 같이 생성하려는 모델의 tree_method 에 'gpu_hist' 만 넣어주면 된다.
그러나 LightGBM 의 경우 GPU 가속을 사용하려면 설치 방법이 비교적 단순하지 않아서 방법을 정리해둔다.
LightGBM 을 빌드하기 위해서 필요한 툴을 아래와 같이 설치한다. 이 때 리눅스 sudo 권한이 필요하다.
빌드 환경이 구축되었다면 아래와 같이, 깃 저장소의 코드를 다운 받아 빌드를 완료한다.
모델 생성할 때 device 에 'gpu'를 주면 된다.
아래를 참고하여 작성하였습니다.
https://lightgbm.readthedocs.io/en/latest/GPU-Tutorial.html
https://stackoverflow.com/questions/60360750/lightgbm-classifier-with-gpu
'노트정리 > 머신러닝 machine leanring' 카테고리의 다른 글
케라스 (keras) 에서 모델 저장 또는 가중치 저장 (1) | 2018.08.24 |
---|---|
케라스(keras) 에서 leaky_relu 쓰기 (0) | 2018.04.01 |