How to install CUDA & cuDNN on Ubuntu 22.04
Install NVIDIA drivers
Update & upgrade
sudo apt update && sudo apt upgrade
Remove previous NVIDIA installation
sudo apt autoremove nvidia* --purge
Check Ubuntu devices
ubuntu-drivers devices
Install drivers
sudo ubuntu-drivers autoinstall
OR
sudo apt install nvidia-driver-535
Reboot & Check
reboot
after restart verify that the following command works
nvidia-smi
Install CUDA drivers
Update & upgrade
sudo apt update && sudo apt upgrade
Install CUDA toolkit
sudo apt install nvidia-cuda-toolkit
This will install nvidia-cuda-toolkit 11.5.
Check CUDA install
nvcc --version
Install cuDNN
Download cuDNN .deb file
You can download cuDNN file here. You will need an Nvidia account. Select the cuDNN version for the appropriate CUDA version, which is the version that appears when you run:
nvcc --version
Install cuDNN
sudo apt install ./<filename.deb>
sudo cp /var/cudnn-<something>.gpg /usr/share/keyrings/
My cuDNN version is 8, adapt the following to your version:
sudo apt update
sudo apt install libcudnn8
sudo apt install libcudnn8-dev
sudo apt install libcudnn8-samples
Test CUDA on Pytorch
Create a virtualenv and activate it
sudo apt-get install python3-pip
sudo pip3 install virtualenv
virtualenv -p py3.10 venv
source venv/bin/activate
Install pytorch
pip3 install torch torchvision torchaudio
Open Python and execute a test
import torch
print(torch.cuda.is_available()) # should be True
t = torch.rand(10, 10).cuda()
print(t.device) # should be CUDA