How to install Pytorch 1.8.0 with cuda support and torchvision

[AastaLLL]

Hi,

Please find the info on below topic:

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.2.0

JetPack 6.0 DP (L4T R3…

JetPack 4
PyTorch v1.8.0

• JetPack 4.4 (L4T R32.4.3) / JetPack 4.4.1 (L4T R32.4.4) / JetPack 4.5 (L4T R32.5.0) / JetPack 4.5.1 (L4T R32.5.1) / JetPack 4.6 (L4T R32.6.1)
  ** Python 3.6 - torch-1.8.0-cp36-cp36m-linux_aarch64.whl

Thanks.