AssertionError: Torch not compiled with CUDA enabled

The links to the wheels are listed in PyTorch for Jetson

https://developer.download.nvidia.cn/compute/redist/jp/v60dp/pytorch/torch-2.2.0a0+6a974be.nv23.11-cp310-cp310-linux_aarch64.whl is correct (with the v60dp part)