Hi, I would like to run my NHITS model which is running on a CPU, now run on a GPU. Are there settings I need to adjust or other things to think/worry about.
m
Marco
04/15/2025, 1:11 PM
Hello! If available, neuralforecast will automatically use the GPU (or TPU in Colab).
I know this is not your question, but if you ever have a GPU but want to run the model on CPU only, then you can specify
accelerator="cpu"
when initializing your model. Otherwise, it automatically goes to the GPU.