Pytorch for JetPack 6.0

[vuthanh.cdt]

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!

[dusty_nv]

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

PyTorch v2.2.0

• JetPack 6.0 DP (L4T R36.2.0)
  • Python 3.10 - torch-2.2.0a0+6a974be.nv23.11-cp310-cp310-linux_aarch64.whl

PyTorch v2.1.0

• JetPack 6.0 DP (L4T R36.2.0)
  • Python 3.10 - torch-2.1.0-cp310-cp310-linux_aarch64.whl (USE_DISTRIBUTED=on)

[hamed.t.g]

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

Thanks

[dusty_nv]

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

[hamed.t.g]

Great, thank you.