Pytorch AND TORCHVISION on AGX Orin

Hi @aif2023, you need to either build torchvision from source (as per the steps under the Installation section of PyTorch for Jetson) or use l4t-pytorch container which already includes PyTorch, torchvision, and torchaudio pre-installed (with CUDA enabled).

And if you have trouble compiling and don’t want to use container, you can copy the pre-compiled PyTorch and torchvision wheels out of l4t-pytorch container and install them natively on your device (the built wheels are found under /opt inside container)