PyTorch version issue with Xavier NX and JetPack 4.4

Hello,
I want to install PyTorch <= 1.5.0 on Jetson Xavier NX. However, with JetPack 4.4 - L4T R32.4.3, it only supports PyTorch==1.6.0. Is there any solutions/workaround?

Many Thanks,
Neel

Hi @neelanjan.bhowmik, due to changes in cuDNN 8.0 from L4T R32.4.3 (specifically the removal of deprecated functions in cuDNN), versions of PyTorch previous to PyTorch 1.6.0 no longer build. You could attempt to identify and backport all of the changes, but it didn’t seem insignificant. It may be easier and time better spent to update to 1.6.0.

Hi,
Possibly, yes, upgrading to 1.6.0 is the best option.
By the way, just curiosity, does Xavier NX support any lower JetPack versions than (<4.4)?

Thanks

JetPack 4.4 Developer Preview (L4T R32.4.2) was the first version of JetPack to support Xavier NX.

JetPack 4.4 Developer Preview (L4T R32.4.2) can run previous PyTorch versions, but unlike L4T R32.4.3 it isn’t a production release.