Install PyTorch 2.0.0 on Jetson Orin

Autonomous Machines Jetson & Embedded Systems Jetson AGX Orin
1

Hi, I have a NVIDIA Orin. Jetpack version of Orin = 5.0.1 .

I wanted to install a pytorch v2.0using virtual env. Therefore, I went to this link: https:/developer.download.nvidia.cn/compute/redist/jp/v50/pytorch and checked but pytorch v2.0is not present.

However, there is av2.0 whl file for Jetpack v5.1.1 in this link:https://developer.download.nvidia.cn/compute/redist/jp/v51/pytorch/

Therefore, I wanted to know if I upgrade my Jetpack to v5.1.1 and use the second link above, will I be able to install pytorch v2.0 successfully ?

Or, if there is any .whl file link for Jetpack 5.0.1 then please share that !

THanks a lot !!

2

Hi @vinayakah ,
Moving this to Jetson Forum for better assistance.

Thanks

3

Sorry for the late response, is this still an issue to support? Thanks

5

Hi,

Please upgrade your device to JetPack 5.1.1 and use the v2.0 PyTorch package.
JetPack 5.0.1 is a DP release, it’s recommended to move to a newer GA release for stability.

Thanks.

7

This topic was automatically closed 14 days after the last reply. New replies are no longer allowed.