LGBMでGPUを使うと遅くなる

Question

以下のページを参考にLGBM-GPUをインストールしました。

http://yutori-datascience.hatenablog.com/entry/2017/07/07/162509

そして、GPUとCPUで速度を検証したとき、CPUのほうが圧倒的に速度が早くなります。

nvidia-smiでGPUの稼働を監視するとちゃんとpythonでGPUが動いているのに速度が改善されません。

この場合、考えられる問題としては何がありえますか？

ご教授いただけると幸いです。

また、どのような情報が必要かわからなかったため、現在の症状しかかけなかったため、必要な情報がありましたら、コメントいただければ追記します。

よろしくお願いいたします。.

Comments

[Q71]

プロファイルを採ってください。近道はありません。ひとつずつ地道に探索してください。

Answer 1

GPUは並列化によって高速で動きますが、コアごとの性能はCPUに劣ります。
そのため、扱っているデータが小さい場合はCPUの方が高速です。
また、max_bin=15程度が最もGPUにとっては高速で、精度を犠牲にgpu_use_dp=falseすることで高速化することはできます。

参考：https://github.com/microsoft/LightGBM/blob/master/docs/GPU-Performance.rst#how-to-achieve-good-speedup-on-gpu

Comments

[退会済みユーザー]

なるほどです。

ありがとうございます。