What PyTorch Version Should I Use for JetPack 6.2.1 on Jetson Orin NX?

Robotics & Edge Computing Jetson Systems Jetson Orin NX
1

Hello,

I have a Jetson Orin NX running JetPack 6.2.1 with L4T 36.4.7. For YOLO apps, I need PyTorch, but the official downloads (https://developer.nvidia.com/embedded/downloads#?search=torch) only support up to JetPack 6.1.

Questions:

  1. Which PyTorch version/source should I use? I’ve seen forum suggestions for PyTorch 2.8.0 – is using the one from https://pypi.jetson-ai-lab.io/jp6/cu126 correct?

  2. Any issues with L4T 36.4.7? Would downgrading to 36.4.4 be better?

5

Hi,

You can install it from the PyTorch official website now:

Please see the updated command below:

Thanks.

6

Hi,

Sorry for the incorrect message above.
Please try the below installation command instead:

$ pip3 install torch torchvision --index-url https://pypi.jetson-ai-lab.io/jp6/cu126 --force-reinstall

Thanks.

8

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