Pytorch with Cuda 11.8 on Jetson ORIN AGX ( aarch64 )

Hi @hg1, you can see the process that I use for building PyTorch wheels in this Dockerfile (there are similar Dockerfiles for building torchvision and torchaudio)

You can basically just follow the ENV variables that I set and the RUN commands that are executed in the shell. There are also build commands for PyTorch and torchvision in PyTorch for Jetson.

Shortly after you begin the PyTorch build, it will print out a configuration summary before it actually starts compiling, and I recommend that you check that summary to make sure the options that you want are enabled (like your desired version of CUDA), before waiting for the entire build to complete just to realize that something didn’t get detected or enabled earlier.