Come installare CUDA su Ubuntu 20.04

CUDA è una piattaforma di elaborazione parallela e un modello di programmazione che offre un'esperienza utente straordinaria quando si sfrutta la GPU per l'elaborazione generica quotidiana. CUDA sta per C componi U nificato D dispositivo A architettura. È stato creato da NVIDIA.

CUDA comprende il toolkit CUDA (compilatore, profilo e debugger), il driver software e CUDA SDK. Il livello software dà accesso diretto al set di istruzioni virtuali della GPU e agli elementi di calcolo paralleli. Per le ricerche di deep learning e gli sviluppatori di framework usa cuDNN per un'accelerazione GPU ad alte prestazioni. È una libreria di primitive per reti neurali profonde.

In questa guida impariamo come installare CUDA e CuDNN su Ubuntu 20.04 . Ci sono due modi per farlo. Puoi installare CUDA da Ubuntu Repository, che è l'approccio più semplice, o installarlo dal repository CUDA che fornisce l'ultima versione di CUDA.

Prerequisiti:

1. Minimo 20 GB di spazio libero su disco
2. 2 GB di RAM consigliati
3. Dovrebbe avere il driver grafico NVIDIA installato. Il driver fornito con CUDA potrebbe non essere l'ultima versione.

Metodo 1:installa CUDA dal repository Ubuntu

Inizia aggiornando gli elenchi dei pacchetti sulla tua istanza Ubuntu 20.04.

$ sudo apt update

Quindi, installa il toolkit CUDA usando il gestore di pacchetti APT come segue. Tieni presente che l'installazione è intensiva e, in quanto tale, assicurati di disporre di una connessione Internet veloce e stabile e di un minimo di 10 G di spazio su disco.

$ sudo apt install nvidia-cuda-toolkit

Una volta completata l'installazione, è necessario aggiungere CUDA a PATH in modo da notificare alla shell la posizione di CUDA. Per fare ciò, specificheremo il PERCORSO nel file .bashrc. Quindi, apri il file usando il tuo editor di testo da riga di comando.

$ sudo vim ~/.bashrc

Alla fine del file, copia e incolla le seguenti righe di codice.

# 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

Quindi, ricarica le modifiche nel file .bashrc.

$ source ~/.bashrc

Ora controlla e conferma la versione di CUDA installata usando il seguente comando.

$ nvcc --version

Metodo 2:installa CUDA dal repository CUDA

Come accennato in precedenza, il repository CUDA fornisce l'ultima versione di CUDA. Quindi, se vuoi installare l'ultima versione, segui le istruzioni qui descritte.

Innanzitutto, scarica il pin del repository CUDA:

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

Quindi, sposta il segnaposto su /etc/preferences.d directory e rinominalo come cuda-repository-pin-600 .

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

Successivamente, scarica il file CUDA Debian. Si noti che il file è piuttosto grande - 2.3G per essere precisi al momento della stesura di questa guida. Assicurati di avere una buona connessione a Internet e ampio spazio su disco.

$ 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

Una volta scaricato, usa il dpkg utility per eseguire il file.

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

Quindi, installa la chiave CUDA GPG come segue:

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

Infine, aggiorna gli elenchi dei pacchetti e installa CUDA utilizzando il gestore di pacchetti APT.

$ sudo apt update

$ sudo apt install cuda -y

Installa CuDNN

CuDNN non viene con Cuda. Per scaricare CuDNN devi registrarti per diventare un membro del programma per sviluppatori NVIDIA, che è gratuito.

Dopo esserti registrato nel programma per sviluppatori NVIDIA, puoi scaricare l'ultima versione di cuDNN. Qui sto scaricando la libreria cuDNN per Linux.

Innanzitutto, estrai il pacchetto cuDNN:

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

Quindi copia i seguenti file nella directory CUDA:

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

Conclusione

In questo tutorial, abbiamo imparato come installare CUDA e cuDNN su Ubuntu 20.04. Grazie per la lettura, per favore lascia il tuo feedback e suggerimenti.