So installieren Sie CUDA unter Ubuntu 20.04

CUDA ist eine Parallel-Computing-Plattform und ein Programmiermodell, das eine bemerkenswerte Benutzererfahrung bietet, wenn GPU für alltägliche Allzweck-Computing genutzt wird. CUDA steht für C rechnen Sie U nifiedD Gerät A Architektur. Es wird von NVIDIA erstellt.

CUDA umfasst das CUDA-Toolkit (Compiler, Profil und Debugger), den Softwaretreiber und das CUDA SDK. Die Softwareschicht bietet direkten Zugriff auf den virtuellen Befehlssatz der GPU und parallele Rechenelemente. Verwenden Sie für Deep-Learning-Forschung und Framework-Entwickler cuDNN für leistungsstarke GPU-Beschleunigung. Es ist eine Bibliothek von Primitiven für tiefe neuronale Netze.

In dieser Anleitung lernen wir, wie man CUDA und CuDNN unter Ubuntu 20.04 installiert . Dazu gibt es zwei Möglichkeiten. Sie können CUDA aus dem Ubuntu-Repository installieren – was der einfachste Ansatz ist – oder aus dem CUDA-Repository installieren, das die neueste Version von CUDA bereitstellt.

Voraussetzungen:

1. Mindestens 20 GB freier Speicherplatz
2. Empfohlen 2 GB RAM
3. Der NVIDIA-Grafiktreiber sollte installiert sein. Der mit CUDA gelieferte Treiber ist möglicherweise nicht die neueste Version.

Methode 1:Installieren Sie CUDA aus dem Ubuntu-Repository

Beginnen Sie damit, die Paketlisten auf Ihrer Ubuntu 20.04-Instanz zu aktualisieren.

$ sudo apt update

Installieren Sie als Nächstes das CUDA-Toolkit mithilfe des APT-Paketmanagers wie folgt. Beachten Sie, dass die Installation intensiv ist, und stellen Sie daher sicher, dass Sie über eine schnelle und stabile Internetverbindung und mindestens 10 GB Speicherplatz verfügen.

$ sudo apt install nvidia-cuda-toolkit

Sobald die Installation abgeschlossen ist, müssen wir CUDA zu PATH hinzufügen, um die Shell über den Speicherort von CUDA zu informieren. Dazu geben wir den PATH in der .bashrc-Datei an. Öffnen Sie die Datei also mit Ihrem Befehlszeilen-Texteditor.

$ sudo vim ~/.bashrc

Kopieren Sie ganz am Ende der Datei die folgenden Codezeilen und fügen Sie sie ein.

# set PATH for cuda 10.1 installation

if [ -d "/usr/local/cuda-10.1/bin/" ]; then

export PATH=/usr/local/cuda-10.1/bin${PATH:+:${PATH}}

export LD_LIBRARY_PATH=/usr/local/cuda-10.1/lib64${LD_LIBRARY_PATH:+:${LD_LIBRARY_PATH}}

fi

Als nächstes laden Sie die Änderungen in die .bashrc-Datei neu.

$ source ~/.bashrc

Überprüfen und bestätigen Sie nun die installierte Version von CUDA mit dem folgenden Befehl.

$ nvcc --version

Methode 2:Installieren Sie CUDA aus dem CUDA-Repository

Wie bereits erwähnt, bietet das CUDA-Repository die neueste Version von CUDA. Wenn Sie also die neueste Version installieren möchten, befolgen Sie die hier beschriebenen Anweisungen.

Laden Sie zuerst den CUDA-Repository-Pin herunter:

$ wget https://developer.download.nvidia.com/compute/cuda/repos/ubuntu2004/x86_64/cuda-ubuntu2004.pin

Verschieben Sie als Nächstes die Stecknadel zu /etc/preferences.d Verzeichnis und benennen Sie es in cuda-repository-pin-600 .

$ sudo mv cuda-ubuntu2004.pin /etc/apt/preferences.d/cuda-repository-pin-600

Laden Sie danach die CUDA-Debian-Datei herunter. Beachten Sie, dass die Datei ziemlich groß ist – 2,3 GB, um genau zu sein, als dieser Leitfaden geschrieben wurde. Stellen Sie sicher, dass Sie über eine gute Internetverbindung und ausreichend Speicherplatz verfügen.

$ wget https://developer.download.nvidia.com/compute/cuda/11.4.2/local_installers/cuda-repo-ubuntu2004-11-4-local_11.4.2-470.57.02-1_amd64.deb

Verwenden Sie nach dem Herunterladen das dpkg Dienstprogramm zum Ausführen der Datei.

sudo dpkg -i cuda-repo-ubuntu2004-11-4-local_11.4.2-470.57.02-1_amd64.deb

Als nächstes installieren Sie den CUDA GPG-Schlüssel wie folgt:

$ sudo apt-key add /var/cuda-repo-ubuntu2004-11-4-local/7fa2af80.pub

Aktualisieren Sie schließlich die Paketlisten und installieren Sie CUDA mit dem APT-Paketmanager.

$ sudo apt update

$ sudo apt install cuda -y

CuDNN installieren

CuDNN wird nicht mit Cuda geliefert. Um CuDNN herunterzuladen, müssen Sie sich als Mitglied des kostenlosen NVIDIA-Entwicklerprogramms registrieren.

Nach der Registrierung im NVIDIA Developer Program können Sie die neueste Version von cuDNN herunterladen. Hier lade ich die cuDNN-Bibliothek für Linux herunter.

Extrahieren Sie zuerst das cuDNN-Paket:

$ tar -xzvf cudnn-10.2-linux-x64-v8.2.4.15.tgz

Kopieren Sie dann die folgenden Dateien in das CUDA-Verzeichnis:

$ 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*

Schlussfolgerung

In diesem Tutorial haben wir gelernt, wie man CUDA und cuDNN unter Ubuntu 20.04 installiert. Vielen Dank fürs Lesen. Bitte hinterlassen Sie Ihr Feedback und Ihre Vorschläge.