sudo apt update && sudo apt upgradesudo apt autoremove nvidia* --purgeubuntu-drivers devicesYou will install the NVIDIA driver whose version is tagged with recommended
sudo ubuntu-drivers autoinstallMy recommended version is 525, adapt to yours
sudo apt install nvidia-driver-525rebootafter restart verify that the following command works
nvidia-smisudo apt update && sudo apt upgradesudo apt install nvidia-cuda-toolkitnvcc --versionYou 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 -V | grep cuda_sudo apt install ./<filename.deb>
sudo cp /var/cudnn-local-repo-ubuntuXXXX.../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-samplessudo apt-get install python3-pip
sudo pip3 install virtualenv
python3 -m venv venv
source venv/bin/activatepip3 install torch torchvision torchaudioimport torch
print(torch.cuda.is_available()) # should be True
t = torch.rand(10, 10).cuda()
print(t.device) # should be CUDA
hi @eliazonta , not sure you are maintaining it, but unfortunately this installs nvida-cuda-toolkit 12.0.
Any hints on how to get nvidia-cuda-toolkit 12.3 installed ?
cheers