Pytorch for JetPack 6.0


I have updated Jetson Orin to Jetpack 6.0, but when I try to install Pytorch with file download from Nvidia “torch-2.1.0a0+41361538.nv23.06-cp38-cp38-linux_aarch64.whl”, it got “ERROR: torch-2.1.0a0+41361538.nv23.06-cp38-cp38-linux_aarch64.whl is not a supported wheel on this platform.”

I think the error from Python 3.10 in Ubuntu 22.04. Have any method to build pytorch with Jetpack 6.0 for Python 3.10? Will it support soon?


Hi @vuthanh.cdt, please see these wheels from PyTorch for Jetson for JetPack 6.0 and Python 3.10:

PyTorch v2.2.0

PyTorch v2.1.0

1 Like

Hi @dusty_nv , Do you have any suggestion for torchvision version compatible with the above mentioned PyTorch versions?


Hi @hamed.t.g, I have been using torchvision 0.16 against these versions of PyTorch wheels.

Great, thank you.

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