https://github.com/nixtla logo
m

marah othman

05/21/2023, 2:47 PM
Does the NeuralForecast library support GPU acceleration on non-NVIDIA GPUs, such as Intel integrated GPUs or AMD GPUs?
f

fede (nixtla) (they/them)

05/22/2023, 8:49 PM
hey @marah othman! We haven’t tested the library on non-nvidia gpus, but pytorch and pytorch-lightning support AMD gpus so it should work :)