alteryx / alteryx/evalml

Add support for GPU acceleration

オープン
#643 コメント 2 件 リアクション 0 件 担当者 0 名 GitHub で見る
needs design new feature
主要言語
Python
スター
850
フォーク
96
PR マージ指標
30日以内にマージされた PR はありません

説明

In the usability blitz, @christopherbunn showed [what looked like an 8x speedup in wall-clock runtime](https://alteryx.quip.com/gwW9AQg5m0Nq/Evalml-Usability-Blitz-April-2020#PFNACAoq60N) when GPU support was enabled in our catboost component. So yes, GPUs are awesome :)

It appears [sklearn has chosen not to prioritize GPU support](https://alteryx.quip.com/gwW9AQg5m0Nq/Evalml-Usability-Blitz-April-2020#PFNACAoq60N). However, [xgboost](https://xgboost.readthedocs.io/en/latest/gpu/), [lightgbm](https://lightgbm.readthedocs.io/en/latest/GPU-Tutorial.html) and [catboost](https://catboost.ai/docs/features/training-on-gpu.html) support GPU acceleration.

Proposed plan:
* Update catboost and xgboost components and pipelines to support turning on GPU acceleration as a boolean flag in the init, default false
* Once the automl strategy project #272 is in, we could write a GPU-specific automl strategy which only runs the pipelines which support GPU acceleration, with it enabled
* That means this is blocked on #272

Questions to consider:
* Should we find more GPU-enabled algorithms?
* Does this vary from platform to platform? We could choose to support only on linux initially.

コントリビューションガイド

コントリビューションガイドを開く

評価

この issue はまだ評価されていません。

新しい issue をメールで受け取る

初心者向けの GitHub issue を短くまとめたダイジェスト。