Hi,
The default CUDA version in JetPack 5.1.2 should be 11.4.
Do you manually upgrade it?
But the latest CUDA for JetPack 5, which use Ubuntu 20.04, is 12.2.
Could you tell us how you install the CUDA 12.4?
We share the prebuilt PyTorch with GPU enabled in the link you shared above.
But you will need to use the default CUDA version.
For other combination, you can build PyTorch from the source to get the package (ex. JetPack 5+CUDA 12.2).
The building instructions can be found in the below topic:
Thanks.