Hi, I would like to run my NHITS model which is ru...
# neural-forecast
j
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
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.
1