Install torch on latest Jetpack 6.2

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

I am having trouble installing torch (with GPU) on my Orin AGX with Jetpack 6.2.

nvcc: NVIDIA (R) Cuda compiler driver
Copyright (c) 2005-2024 NVIDIA Corporation
Built on Wed_Aug_14_10:14:07_PDT_2024
Cuda compilation tools, release 12.6, V12.6.68
Build cuda_12.6.r12.6/compiler.34714021_0

R36 (release), REVISION: 4.3, GCID: 38968081, BOARD: generic, EABI: aarch64, DATE: Wed Jan 8 01:49:37 UTC 2025

KERNEL_VARIANT: oot

TARGET_USERSPACE_LIB_DIR=nvidia
TARGET_USERSPACE_LIB_DIR_PATH=usr/lib/aarch64-linux-gnu/nvidia

4

Hi,

How do you install pytorch?
Please provide the commands step for us to review.

Thanks

5

I tried few things:
sudo python3 -m pip install --no-cache https://developer.download.nvidia.cn/compute/redist/jp/v61/pytorch/torch-2.5.0a0+872d972e41.nv24.08.17622132-cp310-cp310-linux_aarch64.whl

I also tried jp6/cu126 index
and using Python 3.10.12 inside conda, but no luck on getting cuda availability to true.

My Python is 3.12.9.

6

Hi,

Could you follow the https://elinux.org/Jetson/L4T/TRT_Customized_Example#YoloV11_using_Pytorch commands to install torch-2.5?

Thanks