Installing CUDA on a new GPU

Installing NVIDIA Drivers via download

You can select your driver option here: https://www.nvidia.com/download/index.aspx
Then download it as a .run file, it might look something like this: NVIDIA-Linux-x86_64-525.89.02.run
You can run it in a terminal via sh ./path_to_driver/NVIDIA-Linux-x86_64-525.89.02.run
- Note: If you have previously installed NVIDIA drivers, you should remove these first (this may happen when you’re installing the new software, it will ask you if you want to remove the old software).

Run:

sudo apt update
sudo apt upgrade
sudo apt autoremove
sudo apt install nvidia-driver-XXX
- Note: XXX is the number that appears when you run nvidia-smi
- In this case it would be sudo apt install nvidia-driver-525 (notice the 525 in the top left of the output)
Note: Make sure to uninstall existing nvidia-driver-XXX before running (though this may happen automatically when you run the install command.
- E.g. if you install the new drivers via command, it will ask you to remove old ones.
Restart your computer for changes to take effect.

Run this to check what nvidia-* packages you have installed: dpkg -l | grep nvidia
Check out this thread on Stack Overflow for a bunch of ways to troubleshoot existing installations — https://stackoverflow.com/questions/43022843/nvidia-nvml-driver-library-version-mismatch
- I found this answer in particular to remove a bunch of existing NVIDIA packages and then reinstall new ones — https://stackoverflow.com/a/59966222/7900723