Jetson AGX Xavier Pytorch Wheel files for latest Python 3.8/3.9 versions with CUDA 10.2 support

[dusty_nv]

The error indicates that PyTorch wheel was built with support for Nano/TX1 GPU (sm_53) but not the Xavier GPU (sm_72), so it would need rebuilt to include support for sm_72.

You can find the instructions for building PyTorch wheels in the first post in this thread (this is tested against Python 3.6)

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 pre-built l4t-pytorch and l4t-ml container images and Dockerfiles. PyTorch pip wheels JetPack 5

PyTo…

Since you are using AGX Xavier, that is the fastest Jetson to build these wheels on. FYI here is another topic about building PyTorch for Python 3.8:

Install PyTorch with Python 3.8 on Jetpack 4.4.1

Hey @dusty_nv, thank you for your quick response. With your help I was able to build PyTorch 1.7 for Python 3.8 on Jetpack 4.4.1 (Jetson Xavier). It actually helped to switch to the 1.7 branch of PyTorch. Here are the steps that worked for me: Easiest way: git clone --recursive --branch 1.7 http://github.com/pytorch/pytorch cd pytorch python3.8 -m pip install -r requirements.txt python3.8 setup.py install From where I was coming: git clone https://github.com/pytorch/pytorch.git cd pytorch gi…