Need torchvision Wheel for JetPack 5.1.3 on Jetson Orin NX

Robotics & Edge Computing Jetson & Embedded Systems Jetson Orin NX
1

Hello everyone,

I’m currently using a Jetson Orin NX with the following specifications:

  • JetPack Version: 5.1.3
  • CUDA Version: 11.4
  • Python Version: 3.8
  • Installed PyTorch: torch-2.1.0a0+41361538.nv23.06-cp38-cp38-linux_aarch64.whl (working without issues)

I’m looking for a compatible torchvision wheel that works with CUDA 11.4 and PyTorch 2.1.0.

If anyone has a link to the appropriate wheel file or guidance on how to install it, I would greatly appreciate your help!

Thank you!

3

Hi,

We don’t have prebuilt TorchVision for the JetPack 5 user.
Please build it from the source with the instructions below (under Instructions > Installation > torchvision)

Thanks.

5

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