Jetpack 6.2 Pytorch with CUDA support coming any time soon?

Jetson Orin Nano with JetPack 6.2 has CUDA capability sm_87, but no compatible PyTorch wheel exists.

Index of /compute/redist/jp only supports up to version 6.1.

I have had some success in updating my Jetson Toolkit to 12.9 which I though installed these fixes.

My output currently shows these errors.

UserWarning:
Found GPU0 Orin which is of cuda capability 8.7.
PyTorch no longer supports this GPU because it is too old.
The minimum cuda capability supported by this library is 9.0.
warnings.warn(
/home/.local/lib/python3.10/site-packages/torch/cuda/init.py:287: UserWarning:
Orin with CUDA capability sm_87 is not compatible with the current PyTorch installation.
The current PyTorch install supports CUDA capabilities sm_90 sm_100 sm_120.
If you want to use the Orin GPU with PyTorch, please check the instructions at https://pytorch.org/get-started/locally/
warnings.warn(

Can anyone advise on what to do to fix this? I have high hopes for this device, hope we can get it going. I doubt this brand new Jetson Orin Nano SDK is “too old”. Help please! #newb

Hi,

You can find the corresponding package below:

Thanks.

1 Like

Thank you so much! It works!
I used this to install it and it’s working like a champ!
pip3 install torch==2.8.0 torchvision==0.23.0 torchaudio==2.8.0 --index-url jp6/cu126 index

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