PyTorch for Jetson

The latest PyTorch 2.3 wheels for JetPack 6 have been posted, along with wheels for torchvision and torchaudio:

The default version of CUDA in JetPack 6 is CUDA 12.2, so use those unless you have upgraded it to the CUDA 12.4. These were built with jetson-containers.

1 Like