Pytorch Installation CUDA 11.4

Hi, I have a computer with ubuntu 20.04, CUDA 11.4 and NVIDIA drivers 470.
I don’t have the permissions to update it. Which version of Pytorch could I install without having to update the drivers and CUDA?

According to this post I can’t install versions compiled with CUDA 12.1 but I can with CUDA 11.8. Is this correct and why?

Thanks in advance.

Yes, since the PyTorch binaries ship with their own CUDA 11.8 runtime and would use the NVIDIA driver from your system which should be compatible with 11.8.

1 Like