Pytorch Installation CUDA 11.4

1

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 posthttps://discuss.pytorch.org/t/which-pytorch-version-2-0-1-support-cuda-11-4/190446 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.

1 Like
2

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.

2 Likes