Если у вашего драйвера Nvidia CUDA 12.1 не переживайте! Смело ставьте PyTorch под 11.8. Только ни в коем случае не через conda, а через Pip. У меня на получение знания ушла 1 ubuntu 18.04(RIP) и 2 дня.