2017년 4월 25일 화요일

윈도우에 XGBoost 설치하기

딥러닝 세상이지만 케글이나 각종머신러닝 대회에서 XGBoost 역시 꾸준히 좋은 성능을 보여주고 있다. XGBoost의 이론적인 내용은 접어두고 일단 윈도우, 파이썬 환경에서 설치하는 법을 정리해둬야겠다. (다른 PC에 설치할 일이 생길 것 같은 나를 위해서..)

일단 파이썬은 설치되었다고 가정하면 Git과 MinGW가 필요하다.
링크에서 Git 윈도우 버전을 받아 Bash Shell을 실행하면 터미널이 열린다. XGBoost 를 저장할 경로에서 다음의 명령어를 입력한다.

$ git clone --recursive https://github.com/dmlc/xgboost
$ cd xgboost
$ git submodule init
$ git submodule update
cs

복사한 XGBoost를 빌드하려면 64bit 윈도우에서 MinGW-W64가 필요하다.
다운을 받고 설치하는데 setting에서 architecture를 x86_64로 변경한다.

다음 과정이 조금 까다로울 수 있는데 위에서 설치한 MinGW의 라이브러리가 있는 폴더를 윈도우에 등록해줘야 한다.
여기를 참고해 디렉토리(예를 들어 C:\Program Files\mingw-w64\x86_64-5.3.0-posix-seh-rt_v4-rev0\mingw64\bin) 를 시스템 경로로 추가해준다.

Git Bash를 닫고 다시 열어
$ which mingw32-make
cs
를 입력했을 때 위의 디렉토리를 돌려주면 이제 준비 완료.

작업을 편하게 하기 위해 다음을 입력하고
alias make='mingw32-make'
cs
XGBoost를 설치한 곳으로 가 다음을 차례대로 입력해준다.
(XGBoost 튜토리얼은 더 간단한데 그렇게 했을 때 왠지 파이썬에서 설치가 안됐다.)

$ cd dmlc-core
$ make -j4
$ cd ../rabit
$ make lib/librabit_empty.a -j4
$ cd ..
$ cp make/mingw64.mk config.mk
$ make -j4
cs

이제 아나콘다 프롬프트 등에서 XGBoost를 설치하면 된다.

cd code\xgboost\python-package
python setup.py install
cs

파이썬에서 XGBoost를 import 되는 지 다음을 통해 확인하면 진짜 끝.

1
2
3
4
5
6
import os
mingw_path = 'C:\\Program Files\\mingw-w64\\x86_64-5.3.0-posix-seh-rt_v4-rev0\\mingw64\\bin'
os.environ['PATH'= mingw_path + ';' + os.environ['PATH'
import xgboost as xgb
cs

댓글 1개: