Jak zainstalować CUDA 11 (11.6 w tym przypadku) na ubuntu 20.04 przy pomocy "apt install cuda-toolkit-11.6"
sudo apt install cuda-toolkit-11.6
To pozwoli tobie zinstalować CUDA Toolkit 11.6 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 # Old public key #sudo apt-key adv --fetch-keys https://developer.download.nvidia.com/compute/cuda/repos/ubuntu2004/x86_64/7fa2af80.pub # New public key, added 2022-04-25 22:52 sudo apt-key adv --fetch-keys https://developer.download.nvidia.com/compute/cuda/repos/ubuntu2004/x86_64/3bf863cc.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.6
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
Wydaje mi się, że w eksporcie PATH brakuje „bin” na końcu:
export PATH="/usr/local/cuda/bin:$PATH"