The Torchvision compatible with torch 2.0.0+nv23.5 no where to be found on https://forums.developer.nvidia.com/t/pytorch-for-jetson/72048

Autonomous Machines Jetson & Embedded Systems Jetson AGX Orin
1

Hi,
I am struggling to install torch vision for torch-2.1.0a0+41361538.nv23.06-cp38-cp38-linux_aarch64.whl on Jetson Orin Jetpack 5 . I want to run yolov5 on GPU . I followed the PyTorch for Jetson
However, the torch vision wheel is just available for Jetpack 6 and not for Jetpack 5. This is what you see here.

PyTorch v2.1.0

I could not find it in this link neither:
https://download.pytorch.org/whl/torchvision/

I would be grateful if you could Please share the link to torch vision wheel compatible with torch-2.0.0+nv23.05-cp38-cp38-linux_aarch64 or torch-2.1.0a0+41361538.nv23.06-cp38-cp38-linux_aarch64

I am looking forward to hearing from you .

4

Hi,

We only provide TorchVision and TorchAudio for JetPack 6 users.

But you can build it from the source with the instructions listed in the same topic.
Or below container has PyTorch/TorchVision/TorchAudio installed:

Thanks.

1 Like
5

Hi,
Many thanks for the reply and the guidance. I made it work with Jetpack 5.3 ,torch 2.1 and torchvision 0.16 by building it from source.

1 Like
8

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