Compatible pytorch version for the jetson orin with jetpack 6.2

Robotics & Edge Computing Jetson Systems Jetson Orin Nano
1

i am working with jetson orin now and happened to install jetpack 6.2,so now I want to use the gpu in my code,so i want to install pytorch ,but i am not able to find a compatible version for my purpose,so can someone please provide me the answer for this query

3

Hi,

Please find our PyTorch package below:

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

Thanks.

4

hey,thank you for the link,but is the torch and the torchvision compatible,because i get error that says,operator torchvision::nms does’nt exist(for sentence transformers).
Could you please clarify this

5

Hi,

There are some different PyTorch versions on the link.
Please try the below command for the specific steps:

Thanks.

6

hey
i did install pytorch2.5.0,but i am not able to find a compatible torchaudio for this.
let me know if you know the answer
thanks

7

I am facing the same error with PyTorch 2.6.0rc1 and TorchVision 0.20.0 on Jetpack 6.2. Can someone tell me if I should downgrade to Jetpack 6.1 to use gpu in my code or someother Jetpack version?

9

hi,
try jetpack 6.1 as the higher version has few issues

10

Hi, all

Please try below for the packages for JetPack 6.2:

$ pip3 install 'numpy<2'
$ wget https://pypi.jetson-ai-lab.dev/jp6/cu126/+f/6cc/6ecfe8a5994fd/torch-2.6.0-cp310-cp310-linux_aarch64.whl#sha256=6cc6ecfe8a5994fd6d58fb6d6eb73ff2437428bb4953f3ebaa409f83a5f4db99 
$ wget https://pypi.jetson-ai-lab.dev/jp6/cu126/+f/aa2/2da8dcf4c4c8d/torchvision-0.21.0-cp310-cp310-linux_aarch64.whl#sha256=aa22da8dcf4c4c8dc897e7922b1ef25cb0fe350e1a358168be87a854ad114531
$ wget https://pypi.jetson-ai-lab.dev/jp6/cu126/+f/dda/ce98dc7d89263/torchaudio-2.6.0-cp310-cp310-linux_aarch64.whl#sha256=ddace98dc7d892634d2e5b08593436f3e3b1247a1cb11c0d5f4e5ccf64a9be8c 
$ pip3 install --force torch-2.6.0-cp310-cp310-linux_aarch64.whl 
$ pip3 install --force torchvision-0.21.0-cp310-cp310-linux_aarch64.whl 
$ pip3 install --force torchaudio-2.6.0-cp310-cp310-linux_aarch64.whl 

$ python3 
Python 3.10.12 (main, Nov  6 2024, 20:22:13) [GCC 11.4.0] on linux
Type "help", "copyright", "credits" or "license" for more information.
>>> import torch
>>> import torchvision
>>> import torchaudio
>>> torch.__version__
'2.6.0'
>>> torchvision.__version__
'0.21.0'
>>> torchaudio.__version__
'2.6.0'

Thanks.

11

hi,
i get this error while checking

import torchvision
Traceback (most recent call last):
File “”, line 1, in
File “/home/san-jetson-orin/.local/lib/python3.10/site-packages/torchvision/init.py”, line 10, in
from torchvision import _meta_registrations, datasets, io, models, ops, transforms, utils # usort:skip
File “/home/san-jetson-orin/.local/lib/python3.10/site-packages/torchvision/_meta_registrations.py”, line 164, in
def meta_nms(dets, scores, iou_threshold):
File “/home/san-jetson-orin/.local/lib/python3.10/site-packages/torch/library.py”, line 828, in register
use_lib._register_fake(op_name, func, _stacklevel=stacklevel + 1)
File “/home/san-jetson-orin/.local/lib/python3.10/site-packages/torch/library.py”, line 198, in _register_fake
handle = entry.fake_impl.register(func_to_register, source)
File “/home/san-jetson-orin/.local/lib/python3.10/site-packages/torch/_library/fake_impl.py”, line 31, in register
if torch._C._dispatch_has_kernel_for_dispatch_key(self.qualname, “Meta”):
RuntimeError: operator torchvision::nms does not exist

12

Hi,

Could you double-check the version?
The above commands work in our environment.

Thanks.

13

I am now using Jetpack 6.1 and I am now able to use GPU in my code but for some weird reason, I noticed that TensorRT was not present in my Jetpack 6.1

14

Hi, @siddharth.amaranth

Please file a new topic for the new issue.

Thanks.