I want to install PyTorch 2.2.2. The requirement for that
Python 3.8 or later
CUDA 11.8/12.1
The CUDA and Graphics Driver that I have
I am using Ubuntu 20.04.1 LTS (x86_64)
I am not sure the NVIDIA driver version that I have is compatible with CUDA 11.8/12.1?
As you can see in the top right corner of the nvidia-smi output, driver 450 only supports up to cuda 11.0. So you need to upgrade the driver to be able to run anything higher.
Thank you for your answer. I have a follow up question,
How could I find which CUDA version installed in my machine?
If correctly installed, you can use nvcc to display the version
nvcc -V
