개요

Kaggle에서 자주 사용하는 LightGBM을 사용할 때 GPU 가속을 하기 위해 수행해야 하는 빌드 및 설치방법을 정리해둔다.
기본적으로 우분투 16.04에서 정상 동작하니 참고하자.

CUDA설치

CUDA 설치는 우분투 16.04에서 CUDA 성공적으로 설치하기를 참고한다.

우분투 의존성 설치

$ sudo apt-get install --no-install-recommends git cmake build-essential libboost-dev libboost-system-dev libboost-filesystem-dev

LightGBM 빌드

$ git clone --recursive https://github.com/Microsoft/LightGBM
$ cd LightGBM
$ mkdir build ; cd build
$ cmake -DUSE_GPU=1 -DOpenCL_LIBRARY=/usr/local/cuda/lib64/libOpenCL.so -DOpenCL_INCLUDE_DIR=/usr/local/cuda/include/ ..
$ make -j$(nproc)
$ cd ..

위 과정을 수행하면 상위폴더에 lightgbmlib_lightgbm.so가 생성되어 있는것을 확인 할 수 있다.

LightGBM 파이썬 패키지 설치

위에 빌드한 디렉토리에 python-package 디렉토리가 존재한다.
이 디렉토리에 접근하여 아래의 명령어로 파이썬 패키지 설치가 가능하다.

$ pip install setuptools numpy scipy scikit-learn
$ cd python-package/
$ sudo python setup.py install --precompile

GPU 가속 옵션 사용방법

LightGBM 사용시 파라미터로 “device” : “gpu” 를 설정해주면 된다.

참고자료