The latest PyTorch 2.3 wheels for JetPack 6 have been posted, along with wheels for torchvision and torchaudio:
-
JetPack 6.0 (L4T R36.2 / R36.3) + CUDA 12.2
- torch 2.3 -
torch-2.3.0-cp310-cp310-linux_aarch64.whl - torchaudio 2.3 -
torchaudio-2.3.0+952ea74-cp310-cp310-linux_aarch64.whl - torchvision 0.18 -
torchvision-0.18.0a0+6043bc2-cp310-cp310-linux_aarch64.whl
- torch 2.3 -
-
JetPack 6.0 (L4T R36.2 / R36.3) + CUDA 12.4
- torch 2.3 -
torch-2.3.0-cp310-cp310-linux_aarch64.whl - torchaudio 2.3 -
torchaudio-2.3.0+952ea74-cp310-cp310-linux_aarch64.whl - torchvision 0.18 -
torchvision-0.18.0a0+6043bc2-cp310-cp310-linux_aarch64.whl
- torch 2.3 -
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.