How to install Torch and Torchvision with cuda supported in Jetpack 5.1.2

[dusty_nv]

@jonyghosh444 what are the errors that you get when you try building torchvision like under the Installation section of this thread?

PyTorch for Jetson

Below are pre-built PyTorch pip wheel installers for Jetson Nano, TX1/TX2, Xavier, and Orin with JetPack 4.2 and newer. Download one of the PyTorch binaries from below for your version of JetPack, and see the installation instructions to run on your Jetson. These pip wheels are built for ARM aarch64 architecture, so run these commands on your Jetson (not on a host PC). You can also use the containers from jetson-containers. PyTorch pip wheels JetPack 6 ▶ PyTorch v2.3.0 ▶ PyTorch v2.2.0 ▶ PyT…

@jonyghosh444

Then, I tried using a Docker container from this link:
https://github.com/dusty-nv/jetson-containers/tree/master/packages/pytorch.
I used ./autotag, a specific version, docker run, and manually built every process, but in every container, I only got Torch installed, and there was no torchvision installed.

That container only has pytorch - instead use the torchvision container (which includes pytorch + torchvision) or l4t-pytorch (which includes pytorch/torchvision/torchaudio/ect)

• https://github.com/dusty-nv/jetson-containers/tree/master/packages/pytorch/torchvision
• https://github.com/dusty-nv/jetson-containers/tree/master/packages/l4t/l4t-pytorch