Pythonと!

プログラミングド素人の勉強日記です。

Google ColaboratoryでGPU版LightGBMを使う方法

ランタイムの管理でGPUを設定してから、以下のコードを実行

!git clone --recursive https://github.com/Microsoft/LightGBM
%cd /content/LightGBM/
!mkdir build
!cmake -DUSE_GPU=1 #avoid ..
!make -j$(nproc)
!sudo apt-get -y install python-pip
!sudo -H pip install setuptools pandas numpy scipy scikit-learn -U
%cd /content/LightGBM/python-package
!sudo python setup.py install --precompile

ちなみにXgboostは特に設定不要

引用元
GoogleColabratory + GradientBoosting + GPU - Qiita