Hello, Please please may I have a python wheel for pytorch using Jetpack 5 with CUDA 11.8 for python 3.8 on Jetson AGX Xavier arch64. I’m trying to use pytorch and torch audio but torch won’t work as there isn’t a version of pytorch out there compiled for CUDA 1.8 with jetpack 5. I would upgrade J…

Hi, We have ONNXRuntime prebuilt for JetPack. Have you tried it? https://elinux.org/Jetson_Zoo#ONNX_Runtime Thanks.

Jetson AGX Xavier Jetpack 5 upgraded to CUDA 11.8 can't use torch :(

Autonomous Machines Jetson & Embedded Systems Jetson AGX Xavier

miguel.taylor May 2, 2024, 9:26pm 2

Hi,

Jetpack 5 includes CUDA 11.4, and the wheels are compiled for the default CUDA version. If you need a custom CUDA version, you will need to compile Torch from source yourself. You can check the wheels provided by NVIDIA here:

1 Like