Pytorch-CUDA 11.8 for Jetson AGX Orin

@rahulswa if you want everything to be on CUDA 11.8, you would need to rebuild PyTorch with CUDA 11.8 instead. Or build pytorch-3d with CUDA 11.4 instead. If you go with the former, you can find instructions on building PyTorch from source in this thread: