Compatible pytorch version for the jetson orin with jetpack 6.2

Robotics & Edge Computing Jetson & Embedded Systems Jetson Orin Nano
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.

1 Like