PyTorch for Jetson

[dusty_nv]

@huyhung411991

I notice that the patch for Pytorch 1.11 on Jetpack 5.0 doesn’t include the fix for issue #8103. Can you please tell me why?

It’s because I’m working on building Pytorch 1.11/1.12 (with Magma lib) for AGX Xavier & Xavier NX on Jetpack 5.0.2 and also TX2 on Jetpack 4.6.2 and I’m not sure if I have to include that fix or not.

Hi @huyhung411991, the JetPack 5.x wheels are for Xavier & Orin, and I’ve only seen that issue on TX1/TX2/Nano. So to get the full perf on Xavier & Orin, I did not include it. However if you do encounter the issue on JetPack 5, please let me know - thanks.