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

スポンサーリンク
processor 全般

みなさんこんにちは!

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

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

コード

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

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

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

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

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

それでは!

コメント

タイトルとURLをコピーしました