Different Cuda toolkit version between PC for training model and Jetson AGX Orin

Hi,

Suppose yes.

CUDA 12.8 is also available on the AGX Orin.
Please setup your environment with JetPack 6.2 and upgrade the CUDA with the below link:

Thanks.