Pytorch for JetPack 6.0

Robotics & Edge Computing Jetson Systems Jetson Orin NX
1

Hi,

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?

Thanks!

2

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

5

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

Thanks

6

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

7

Great, thank you.