PyTorch for Jetpack 6.0

[nitin_sk]

How should I install PyTorch for Jetpack 6.0

[AastaLLL]

Hi,

Please check the info in the below link:

Pytorch for JetPack6.0

Hello, I would like to know if there are PyTorch pip wheels available for JetPack 6.0 (L4T R36.3.0). I couldn’t find PyTorch for the latest version at the provided link.

Thanks.

[nitin_sk]

The PyTorch link which you have provided is for JetPack 6.0 DP, I need it for 6.0

[dusty_nv]

@nitin_sk they are the same wheels, because the CUDA version is the same (12.2)

I updated this announcement thread yesterday with the latest for PyTorch 2.3 on JetPack 6:

PyTorch for Jetson

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 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.…