https://github.com/nixtla logo
s

SamT

04/24/2023, 11:12 PM
Hello team, Quick question: the newest version of neuralforecast has a runtime dependency on Pytorch 2.0.0 (and the corresponding lightning package). Is this necessary, or can I still use 1.13? There are no tests to verify if my installation will work with an older version like that. (Tbh I only want maybe 1.12 or 1.13 support, nothing older).
f

fede (nixtla) (they/them)

04/27/2023, 3:23 AM
hey @SamT! Yes, neuralforecast requires pytorch 2.0 since the newest version of ligthning only supports that version. An option would be to fix the neuralforecast version, including torch:
neuralforecast<1.5.0 torch<2.0.0
3 Views