Jak zjistit verzi CUDA v Ubuntu [Jednoduchý návod s příkazy]

1. Úvod

CUDA (Compute Unified Device Architecture) je platforma pro paralelní výpočty vyvinutá společností NVIDIA, která využívá GPU. Je široce používána pro strojové učení, hluboké učení, 3D renderování a různé další výpočetní úlohy.

Při používání CUDA v prostředí Ubuntu je zásadní zkontrolovat verzi CUDA z následujících důvodů:

Kompatibilita ovladače

CUDA vyžaduje konkrétní verzi ovladače NVIDIA, aby fungovala správně. Pokud jsou verze nekompatibilní, CUDA nemusí pracovat správně.

Kompatibilita knihoven

Knihovny jako TensorFlow a PyTorch vyžadují specifické verze CUDA a cuDNN. Je nezbytné zajistit, že máte nainstalovanou odpovídající verzi.

Předejití záměně v systému

Pokud je v systému nainstalováno více verzí CUDA, je nutné zjistit, která verze je aktivní, a podle potřeby mezi nimi přepínat.

V tomto článku poskytneme jasné vysvětlení jak zkontrolovat verzi CUDA v Ubuntu.

2. Jak zkontrolovat verzi CUDA v Ubuntu

V prostředí Ubuntu můžete verzi CUDA zkontrolovat následujícími metodami:

Metoda 1: Kontrola pomocí příkazu nvidia-smi (nejjednodušší metoda)

Ovladač NVIDIA obsahuje nástroj nazvaný nvidia-smi (NVIDIA System Management Interface), který umožňuje zjistit stav vašeho GPU.

Příkaz k provedení

nvidia-smi

Příklad výstupu

+-----------------------------------------------------------------------------+
| NVIDIA-SMI 530.41.03    Driver Version: 530.41.03    CUDA Version: 12.1     |
+-----------------------------------------------------------------------------+

Klíčové body

• Zobrazená CUDA Version: 12.1 představuje maximální verzi CUDA podporovanou ovladačem NVIDIA.
• Tato verze nemusí vždy odpovídat nainstalované verzi CUDA toolkit, proto se doporučuje provést kontrolu i dalšími metodami.

Metoda 2: Kontrola pomocí příkazu nvcc -V (pro vývojáře)

Pokud je CUDA správně nainstalována, můžete zkontrolovat verzi nvcc (kompilátoru CUDA).

Příkaz k provedení

nvcc -V

Příklad výstupu

nvcc: NVIDIA (R) Cuda compiler driver
Copyright (c) 2005-2023 NVIDIA Corporation
Built on Sun_Jul_30_19:09:40_PDT_2023
Cuda compilation tools, release 12.1, V12.1.105

Klíčové body

• Část, která uvádí release 12.1, V12.1.105, označuje nainstalovanou verzi CUDA toolkit.
• Tato verze se může lišit od verze zobrazené pomocí nvidia-smi, proto buďte opatrní.

Metoda 3: Kontrola souboru version.txt (manuální ověření)

Pokud je CUDA nainstalována v /usr/local/cuda, informace o verzi jsou zaznamenány v souboru version.txt.

Příkaz k provedení

cat /usr/local/cuda/version.txt

Příklad výstupu

CUDA Version 12.1.105

Klíčové body

• Tato metoda je užitečná, pokud není k dispozici příkaz nvcc -V.
• Ujistěte se, že /usr/local/cuda je správně propojen s požadovanou verzí CUDA.

3. Jak zkontrolovat verzi cuDNN

cuDNN (CUDA Deep Neural Network) je knihovna určená pro hluboké učení a používá se v kombinaci s CUDA.
Kromě kontroly verze CUDA je také důležité ověřit verzi cuDNN.

Metoda 1: Kontrola souboru cudnn_version.h

Verze cuDNN je zaznamenána v hlavičkovém souboru cudnn_version.h.

Příkaz k provedení

cat /usr/local/cuda/include/cudnn_version.h | grep CUDNN_MAJOR -A 2

Příklad výstupu

#define CUDNN_MAJOR 8
#define CUDNN_MINOR 9
#define CUDNN_PATCHLEVEL 1

Klíčové body

• Tento výstup potvrzuje, že je nainstalována cuDNN 8.9.1.
• Použití příkazu grep vám umožní snadno získat informace o verzi cuDNN.
• Protože cuDNN musí být kompatibilní s CUDA, je důležité ověřit správnou kombinaci verzí.

Metoda 2: Kontrola pomocí příkazu dpkg (pro Debian‑based Linux)

Na Ubuntu a dalších distribucích založených na Debianu můžete zkontrolovat nainstalovanou verzi cuDNN pomocí příkazu dpkg.

Příkaz k provedení

dpkg -l | grep libcudnn

Ukázkový výstup

ii  libcudnn8    8.9.1-1+cuda12.1    amd64    NVIDIA cuDNN Library

Klíčové body

• Část libcudnn8 8.9.1-1+cuda12.1 potvrzuje nainstalovanou verzi cuDNN (8.9.1) .
• Část cuda12.1 označuje kompatibilní verzi CUDA (12.1) .

Použitím těchto metod můžete zajistit, že je vaše prostředí CUDA správně nakonfigurováno.

4. Jak zvládnout více nainstalovaných verzí CUDA

V prostředí Ubuntu lze nainstalovat více verzí CUDA. To však může někdy vést k záměně, která verze je aktuálně aktivní.
V takových případech je třeba přepnout na odpovídající verzi.

Metoda 1: Přepnutí pomocí update-alternatives

V Ubuntu můžete použít update-alternatives k přepínání verzí CUDA.

Zkontrolujte aktuální nastavení

update-alternatives --query cuda

Přepněte verzi CUDA

sudo update-alternatives --config cuda

Ukázkový výstup

There are 3 choices for the alternative cuda (providing /usr/local/cuda).

  Selection    Path                Priority   Status
------------------------------------------------------------
* 0            /usr/local/cuda-11.8  100       auto mode
  1            /usr/local/cuda-10.2  50        manual mode
  2            /usr/local/cuda-11.8  100       manual mode
  3            /usr/local/cuda-12.1  110       manual mode

Press <enter> to keep the current choice[*], or type selection number:

Klíčové body

• Spuštěním update-alternatives --config cuda se zobrazí seznam dostupných verzí CUDA.
• Po zadání odpovídajícího čísla můžete vybrat požadovanou verzi CUDA.
• K dispozici jsou auto režim a manual režim; zvolte manual režim, pokud chcete verze přepínat ručně.

Metoda 2: Ruční nastavení symbolického odkazu

Verze CUDA můžete také přepínat úpravou symbolického odkazu.

Zkontrolujte existující symbolický odkaz

ls -l /usr/local/cuda

Ukázkový výstup

lrwxrwxrwx 1 root root 20 Feb  1 12:34 /usr/local/cuda -> /usr/local/cuda-11.8

Změňte verzi CUDA

sudo rm /usr/local/cuda
sudo ln -s /usr/local/cuda-12.1 /usr/local/cuda

Ověřte změnu

ls -l /usr/local/cuda

Klíčové body

• /usr/local/cuda slouží jako výchozí cesta CUDA. Změna tohoto odkazu přepne verzi CUDA.
• Pomocí příkazu ln -s můžete snadno změnit verzi CUDA, aniž byste museli upravovat systémové konfigurace.

Pomocí těchto metod můžete efektivně spravovat více verzí CUDA a zajistit, že používáte správnou verzi podle svých potřeb.

5. Často kladené otázky (FAQ)

Zde jsou některé časté otázky týkající se kontroly verze CUDA. Pokud narazíte na problémy, podívejte se na tato řešení.

Q1: Příkaz nvcc -V nebyl nalezen!

Pokud příkaz nvcc není nalezen, CUDA možná není správně nainstalována, nebo není nastavena její cesta.

Řešení 1: Zkontrolujte, zda je CUDA nainstalována

ls /usr/local/cuda/

Řešení 2: Přidejte nvcc do cesty

export PATH=/usr/local/cuda/bin:$PATH
export LD_LIBRARY_PATH=/usr/local/cuda/lib64:$LD_LIBRARY_PATH

Po spuštění těchto příkazů zkuste znovu spustit nvcc -V, abyste zjistili, zda se verze zobrazí správně.

Q2: Proč se verze CUDA zobrazená pomocí nvidia-smi liší?

Verze CUDA zobrazovaná pomocí nvidia-smi představuje maximální verzi CUDA podporovanou ovladačem NVIDIA, ne nutně nainstalovanou verzi nástroje CUDA.

Jak zkontrolovat:

nvidia-smi

Ukázkový výstup:

CUDA Version: 12.1

Pro kontrolu skutečně nainstalované verze CUDA použijte nvcc -V nebo zkontrolujte soubor version.txt.

Q3: Jak zkontrolovat kompatibilitu CUDA a cuDNN?

Nejlepším způsobem, jak zkontrolovat kompatibilitu mezi CUDA a cuDNN, je nahlédnout do oficiální matice podpory od NVIDIA.

Oficiální dokumentace:

NVIDIA cuDNN Support Matrix

Kromě toho můžete zkontrolovat nainstalované verze pomocí následujících příkazů:

Zkontrolovat verzi CUDA

nvcc -V

Zkontrolovat verzi cuDNN

cat /usr/local/cuda/include/cudnn_version.h | grep CUDNN_MAJOR -A 2

Správnou správou vašeho prostředí můžete předejít problémům s kompatibilitou CUDA a cuDNN.

6. Shrnutí

V tomto článku jsme vysvětlili, jak zkontrolovat verzi CUDA v prostředí Ubuntu.
Pojďme si zopakovat hlavní body.

Způsoby, jak zkontrolovat verzi CUDA

Způsoby, jak zkontrolovat verzi cuDNN

Jak přepínat verze CUDA

Klíčové poznatky

• Je důležité správně identifikovat verzi CUDA
• Zajistěte kompatibilitu mezi CUDA a cuDNN
• Pokud používáte více verzí CUDA, pochopte, jak mezi nimi přepínat

Správnou správou vašeho prostředí můžete maximalizovat výhody CUDA.
Doufáme, že vám tento článek pomůže zkontrolovat verzi CUDA ve vašem Ubuntu prostředí.

Související články