alteryx / alteryx/evalml

Add support for GPU acceleration

Aperta
#643 2 commenti 0 reazioni 0 assegnatari Vedi su GitHub
needs design new feature
Lingua principale
Python
Stelle
850
Fork
96
Metriche di merge delle PR
Nessuna PR unita negli ultimi 30g

Descrizione

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.

Guida per i contributori

Apri la guida per i contributori

Valutazione

Questa issue non è ancora stata valutata.

Ricevi le nuove issue nella tua casella

Un breve riepilogo di issue GitHub adatte ai principianti.