Torch.cuda.is_available() is False for CUDA version 11.4

For Jetson you would have to use the custom wheels from NVIDIA which you can find e.g. here and here.