Pytorch CUDA for Jetson orin nano

I want to use my Jetson orin nano to train YOLO but it can’t use GPU for training.
My Jetson is installed with Jetpack 6.2 and CUDA 12.6. Is there pytorch version that compatible with this setup?

You can find the wheels for JetPack6 here.