Which Cuda and pytroch with Jetpack 6 on Jetson agx orin dev kit

Robotics & Edge Computing Jetson Systems Jetson AGX Orin
,
1

Hi there,

I am trying to build object detection with the zed x mini camera on my jetson agx orin dev kit. For this I have a trained pytorch model that I want to use. But now I have the problem that I cannot find a working combination of Jetpack, Cuda and Pytorch that works on my Jetson.
On pytorch.org it says that I need Cuda 11.8, 12.1 or 12.4.
Since I have Jetpack 6 and Ubuntu 22.04, I cannot find a download link for Cuda 11.8 and 12.1.
On the link below I cannot configure my system for one of the required Cuda versions because of my architecture. Can I use a different architecture and get it to work somehow?
Or do you guys have any other idea to make it work?
CUDA Toolkit 12.1 Downloads | NVIDIA Developer

Thank you very much!

4

Hi,

If you are using JetPack 6.1, please find below the ML-related packages for Jetson.
http://jetson.webredirect.org/jp6/cu126

For JetPack 6.0, you can also find the PyTorch/TorchVision/TorchAudio in the below topic:

Thanks.