【1行】Google ColabでLightGBMをGPUで動かす

みなさんこんにちは！

今回もGoogle Colabで〇〇をGPUで動かすシリーズです。

↑LightGBMの新しいバージョンではインストール方法が変更になりました！↑

コード

!pip uninstall --yes lightgbm && pip install --install-option=--gpu lightgbm

これでインストールをしたら、max_depthなどのパラメタと一緒に、

params = {
    ....
    "device": "gpu", 
    ....
}

とすればGPUで動きます！学習も早くなるはず。

ランタイムをGPUに変更するのも忘れずに！

それでは！