PyTorch for Jetson

Robotics & Edge Computing Jetson Systems Jetson Orin NX
1

Hi Team,

I am trying train the yolo model for object detection and I visited the below mentioned website to install the torch and torch-vision package.
Link:- PyTorch for Jetson

But in the mentioned link no torch and torch vision package mentioned to install which supports for jetpack 6.2 (L4T 36.4.3 installed on Jetson Orin NX SoM Module.

Please provide the solution to install the torch and torch vision package.

@WayneWWW
@DaneLLL

3

Moving your topic to the OrinNX board (here is for Thor).

Hi,

Please find the packages for JetPack 6.1+ in the link below:

Thanks.

4

Hi,

We are not able to open the link.

@AastaLLL

5

Hi,
The link https://pypi.jetson-ai-lab.io/jp6/cu126 opens inconsistently — sometimes it works, sometimes it doesn’t.
Please check or share an alternate link.

@AastaLLL

6

Hi @AastaLLL,

I am following the below mentioned Query to Install torch and torch vision which supports CUDA for Jetson Orin-NX platform flashed with L4T 36.4.3 and cuda 12.6.

Query Link:- PyTorch for Jetson

In the above Query Link I have tried to download torch and torch vision package which supports the 12.6 cuda and jetpack 6.2.1

but Link is not opening which you have mentioned,
https://pypi.jetson-ai-lab.io/jp6/cu126

Requesting your help to download the Torch and torch-vision for jetpack 6.2.1

Thanks,
ChaithraBV

7

Hi,

Sorry that the server is up now.
Please try it again.

Thanks.

1 Like
8

Thankyou @AastaLLL

The Link is up now

10

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