GPU Training

In this tutorial, we will learn how to use GPU for training.

Using GPU for Training

To use GPU for training, specify DEVICE in a config file.

# gpu-training.yml
DEVICE: "cuda:0"

Initialize TimeSeriesForecaster with the config file.

from tsts.solvers import TimeSeriesForecaster

forecaster = TimeSeriesForecaster("gpu-training.yml")