Chad Williamson
07/15/2023, 10:42 PMCristian (Nixtla)
07/16/2023, 5:31 PMcuda
is available for your PyTorch installation. Installing directly with pip
or conda
should install everything correctly. You can check that everything works properly with torch.cuda.is_available()
, it should return TrueMax (Nixtla)
07/16/2023, 6:12 PMtorch.cuda.is_available()
returns true, then GPU should be used automatically by NeuralForecast. You don't have to configure anything extra.Chad Williamson
07/16/2023, 8:28 PM