Using Pytorch on GPU after upgrading cuda to 11.8 on Jetpack5

@YoushaaMurhij you might want to try re-installing the PyTorch 2.0 wheel that has CUDA support from here:

Or baring that, rebuild PyTorch from source for CUDA 11.8 following the Build from Source instructions from that thread. I haven’t tried upgrading CUDA with PyTorch before, so YMMV. Also if/when you go to build torchvision, the CUDA versions will need to match otherwise PyTorch will throw an error about it.