Skip to content
New issue

Have a question about this project? Sign up for a free GitHub account to open an issue and contact its maintainers and the community.

By clicking “Sign up for GitHub”, you agree to our terms of service and privacy statement. We’ll occasionally send you account related emails.

Already on GitHub? Sign in to your account

[FEATURE REQUEST] Add GPU/CPU Multi-Core Processing for each forecasting method. #146

Open
muazhari opened this issue Feb 5, 2023 · 1 comment

Comments

@muazhari
Copy link

muazhari commented Feb 5, 2023

Is your feature request related to a problem? Please describe.
I'm always frustrated when trying to train forecasting models because it is slow and not using all my CPU cores.

Describe the solution you'd like
Add GPU/CPU Multi-Core Processing for each forecasting method.

Describe alternatives you've considered
A clear and concise description of any alternative solutions or features you've considered.

Additional context
Add any other context or screenshots about the feature request here.

@muazhari muazhari changed the title [FEATURE REQUEST] Add Multi-Core Processing for each forecasting method. [FEATURE REQUEST] Add GPU / CPU Multi-Core Processing for each forecasting method. Feb 5, 2023
@muazhari muazhari changed the title [FEATURE REQUEST] Add GPU / CPU Multi-Core Processing for each forecasting method. [FEATURE REQUEST] Add GPU/CPU Multi-Core Processing for each forecasting method. Feb 5, 2023
@aadyotb
Copy link
Contributor

aadyotb commented Feb 14, 2023

Thanks for the comment. As far as multi-CPU support goes, the underlying API supports it for some models (Prophet, tree models, random cut forest, isolation forest; refer to API documentation for each model), but not others (ARIMA, ETS, spectral residual). GPU is already supported for all deep learning models, and the model will automatically use the GPU if one is available.

We are also investigating GPU support for other models like ARIMA and tree models using Rapids, but this is further out on the roadmap. Let me know if this addresses your questions.

Sign up for free to join this conversation on GitHub. Already have an account? Sign in to comment
Labels
None yet
Projects
None yet
Development

No branches or pull requests

2 participants