Instalacja CUDA 11 na ubuntu 20.4

Jak zainstalować CUDA 11 (11.3 w tym przypadku) na ubuntu 20.04 przy pomocy jednego polecenia

sudo apt install cuda-toolkit-11.1

To pozwoli tobie zinstalować CUDA Toolkit 11.3 wraz z narzędziami (nvcc), bibliotekami do linkowania i resztą majdanu. Jeżeli chcesz iść na skróty to cały skrypt do instalacji CUDA 11 na ubuntu 20.04 możesz zobaczyć na moim gist.

Sprawdź, czy system widzi twoją kartę graficzną

Wydając polecenie lspci upewnij się że system poprawnie widzi twoją kartę.

lspci | grep -i nvidia

Powinienś otrzymać wydruk na konsoli podobny do tego poniżej. Uwaga ja na serwerze mam 4 karty RTX 3090 więc wydruk jest większy

03:00.0 VGA compatible controller: NVIDIA Corporation Device 2204 (rev a1)
03:00.1 Audio device: NVIDIA Corporation Device 1aef (rev a1)
21:00.0 VGA compatible controller: NVIDIA Corporation Device 2204 (rev a1)
21:00.1 Audio device: NVIDIA Corporation Device 1aef (rev a1)
49:00.0 VGA compatible controller: NVIDIA Corporation Device 2204 (rev a1)
49:00.1 Audio device: NVIDIA Corporation Device 1aef (rev a1)
4a:00.0 VGA compatible controller: NVIDIA Corporation Device 2204 (rev a1)
4a:00.1 Audio device: NVIDIA Corporation Device 1aef (rev a1)

Zainstaluj sterowniki do karty graficznej

Jeżeli już wcześniej instalowałeś cuda i sterowniki to usuń je.

sudo apt-get purge nvidia*
sudo apt remove nvidia-*
sudo rm /etc/apt/sources.list.d/cuda*
sudo apt-get autoremove && sudo apt-get autoclean
sudo rm -rf /usr/local/cuda*

Następnie zainstaluj nowe sterowniki do twojej karty graficznej.

sudo add-apt-repository ppa:graphics-drivers/ppa
sudo apt update
sudo apt install nvidia-driver-495

Dodaj NVidia Ubuntu 20.04 repository for CUDA 11

Zaczynamy od dodania oficjalnego repozytorium od NVidia z całym środowiskiem.

wget https://developer.download.nvidia.com/compute/cuda/repos/ubuntu2004/x86_64/cuda-ubuntu2004.pin
sudo mv cuda-ubuntu2004.pin /etc/apt/preferences.d/cuda-repository-pin-600
sudo apt-key adv --fetch-keys https://developer.download.nvidia.com/compute/cuda/repos/ubuntu2004/x86_64/7fa2af80.pub
sudo add-apt-repository "deb http://developer.download.nvidia.com/compute/cuda/repos/ubuntu2004/x86_64/ /"

Instalujemy CUDA Toolkit przy pomocy apt

Po dodaniu repozytorium z możesz przy pomocy narzędzia apt install łatwo i szybko zainstalować tę wersję, która jest tobie potrzebna.

sudo apt update
sudo apt install cuda-toolkit-11.1

Opcjonalnie na potrzeby innych bibliotek do ML zainstaluj cuDNN

Ściągnij bibliotekę cuDNN ze strony NVidia. Musisz się zalogować (stworzyć konto jeżeli nie masz), odpowiedzieć na kilka prostych ptyań i wtedy zostaniesz przekierowany do strony z ściąganiem. Znajdź odpowiednią wersję dla twojego systemu, wybierz najnowszą wersję biblioteki i zapisz na komputerze

tar -xzvf cudnn-11.2-linux-x64-v8.1.1.33.tgz

sudo cp cuda/include/cudnn*.h /usr/local/cuda/include
sudo cp cuda/lib64/libcudnn* /usr/local/cuda/lib64
sudo chmod a+r /usr/local/cuda/include/cudnn*.h /usr/local/cuda/lib64/libcudnn*

Dodaj CUDA_HOME do zmiennej środowiskowej PATH

Wyedytuj plik  /home/$USER/.bashrc i dodaj wpis na końcu pliku

export LD_LIBRARY_PATH="$LD_LIBRARY_PATH:/usr/local/cuda/lib64:/usr/local/cuda/extras/CUPTI/lib64"
export CUDA_HOME=/usr/local/cuda
export PATH="/usr/local/cuda/:$PATH"

Photo by Dmitriy Demidov on Unsplash