Pytorch for JetPack6.0

[643870525]

Hello, I would like to know if there are PyTorch pip wheels available for JetPack 6.0 (L4T R36.3.0).

PyTorch for Jetson

Below are pre-built PyTorch pip wheel installers for Jetson Nano, TX1/TX2, Xavier, and Orin with JetPack 4.2 and newer. Download one of the PyTorch binaries from below for your version of JetPack, and see the installation instructions to run on your Jetson. These pip wheels are built for ARM aarch64 architecture, so run these commands on your Jetson (not on a host PC). You can also use the containers from jetson-containers. PyTorch pip wheels JetPack 6

PyTorch v2.2.0

JetPack 6.0 DP (L4T R3…

I couldn’t find PyTorch for the latest version at the provided link.

[AastaLLL]

Hi,

The package is released in a monthly manner.
Please wait for the release.

Thanks.

[dusty_nv]

@643870525 the ones built on R36.2.0 work on R36.3.0, because they use the same CUDA 12.2 and cuDNN 8.9.

I do have PyTorch 2.3 built awhile against CUDA 12.4 and cuDNN 9 on R36.3.0: http://jetson.webredirect.org/jp6/cu124

I will also go back and build PyTorch 2.3 for CUDA 12.2 on R36.3.0.

[dusty_nv]

OK, you can find PyTorch 2.3 built on R36.3 for CUDA 12.2 here: jp6/cu122/: torch-2.3.0 metadata and description