JetPack 6.1 install Pytorch 2.5.1

Robotics & Edge Computing Jetson & Embedded Systems Jetson Orin Nano
1

where is new pytorch version 2.5.1 for jetpack6.1? Download center contains just offers pytorch v2.5.0a0.

Thanks in advance!

3

Hi,

Does PyTorch 2.6 work for you?
The latest prebuilt for JetPack 6.1 is 2.6.0rc1:

https://pypi.jetson-ai-lab.dev/jp6/cu126

Thanks.

4

Unfortunately, v2.6 seems to be incompatible with Numpy, crashing at runtime :(.
Additionally, I see that torchvision and torchaudio seem to be incompatible with it as well.

5

I have the same question because when using this I found issues with the CuDNN with 2.6.0rc1 so is this actually the correct working version with Jetpack 6.1 and python 3.10?

6

You will have to build it your self and it does run on jp6.1

py-6-1
py-6-11498×507 147 KB

You can build all of it natively on the board, no need to CC.

1 Like
7

Hi,

We provide 2.4.0, 2.5.0, and 2.6.0rc1 for the JetPack 6.1 environment.
https://pypi.jetson-ai-lab.dev/jp6/cu126/torch

For other versions, you can build it from the source.
You can find the building command in the below topic:

Thanks.

1 Like
9

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