Jak sprawdzić wersję CUDA w Ubuntu [Łatwy przewodnik po poleceniach]

1. Wprowadzenie

CUDA (Compute Unified Device Architecture) to platforma obliczeń równoległych opracowana przez NVIDIA, wykorzystująca procesory graficzne (GPU).
Jest szeroko stosowana w uczeniu maszynowym, uczeniu głębokim, renderowaniu 3D oraz w różnych innych zadaniach obliczeniowych.

Podczas korzystania z CUDA w środowisku Ubuntu, kluczowe jest sprawdzenie wersji CUDA z następujących powodów:

Zgodność sterownika

CUDA wymaga określonej wersji sterownika NVIDIA, aby działać prawidłowo. Jeśli wersje są niezgodne, CUDA może nie działać poprawnie.

Zgodność bibliotek

Biblioteki takie jak TensorFlow i PyTorch wymagają konkretnych wersji CUDA i cuDNN. Należy upewnić się, że zainstalowano odpowiednią wersję.

Zapobieganie zamieszaniu w systemie

Jeśli w systemie zainstalowano wiele wersji CUDA, konieczne jest zidentyfikowanie, która wersja jest aktywna i przełączanie się między wersjami w razie potrzeby.

W tym artykule przedstawimy jasne wyjaśnienie jak sprawdzić wersję CUDA w Ubuntu.

2. Jak sprawdzić wersję CUDA w Ubuntu

W środowisku Ubuntu możesz sprawdzić wersję CUDA przy użyciu następujących metod:

Metoda 1: Sprawdzenie poleceniem nvidia-smi (najprostsza metoda)

Sterownik NVIDIA zawiera narzędzie o nazwie nvidia-smi (NVIDIA System Management Interface), które umożliwia sprawdzenie statusu Twojego GPU.

Polecenie

nvidia-smi

Przykładowy wynik

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

Kluczowe informacje

• CUDA Version: 12.1 wyświetlona tutaj reprezentuje maksymalną wersję CUDA obsługiwaną przez sterownik NVIDIA.
• Może ona nie zawsze odpowiadać zainstalowanej wersji zestawu narzędzi CUDA, dlatego zaleca się sprawdzenie przy użyciu dodatkowych metod.

Metoda 2: Sprawdzenie poleceniem nvcc -V (dla programistów)

Jeśli CUDA jest poprawnie zainstalowana, możesz sprawdzić wersję nvcc (kompilatora CUDA).

Polecenie

nvcc -V

Przykładowy wynik

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

Kluczowe informacje

• Fragment release 12.1, V12.1.105 wskazuje zainstalowaną wersję zestawu narzędzi CUDA.
• Może się różnić od wersji wyświetlanej przez nvidia-smi, więc zachowaj ostrożność.

Metoda 3: Sprawdzenie pliku version.txt (weryfikacja ręczna)

Jeśli CUDA jest zainstalowana w /usr/local/cuda, informacje o wersji są zapisane w pliku version.txt.

Polecenie

cat /usr/local/cuda/version.txt

Przykładowy wynik

CUDA Version 12.1.105

Kluczowe informacje

• Ta metoda jest przydatna, jeśli polecenie nvcc -V jest niedostępne.
• Upewnij się, że /usr/local/cuda jest poprawnie powiązany z żądaną wersją CUDA.

3. Jak sprawdzić wersję cuDNN

cuDNN (CUDA Deep Neural Network) to biblioteka przeznaczona do uczenia głębokiego, używana w połączeniu z CUDA.
Oprócz sprawdzania wersji CUDA, ważne jest również zweryfikowanie wersji cuDNN.

Metoda 1: Sprawdzenie pliku cudnn_version.h

Wersja cuDNN jest zapisana w pliku nagłówkowym cudnn_version.h.

Polecenie

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

Przykładowy wynik

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

Kluczowe informacje

• Ten wynik potwierdza, że zainstalowano cuDNN 8.9.1.
• Użycie polecenia grep umożliwia łatwe pobranie informacji o wersji cuDNN.
• Ponieważ cuDNN musi być zgodny z CUDA, ważne jest zweryfikowanie prawidłowej kombinacji wersji.

Metoda 2: Sprawdzenie poleceniem dpkg (dla systemów Linux opartych na Debianie)

W Ubuntu i innych dystrybucjach Linux opartych na Debianie możesz sprawdzić zainstalowaną wersję cuDNN przy użyciu polecenia dpkg.

Polecenie wykonania

dpkg -l | grep libcudnn

Przykładowy wynik

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

Najważniejsze informacje

• Fragment libcudnn8 8.9.1-1+cuda12.1 potwierdza zainstalowaną wersję cuDNN (8.9.1) .
• Fragment cuda12.1 wskazuje kompatybilną wersję CUDA (12.1) .

Korzystając z tych metod, możesz zapewnić prawidłową konfigurację środowiska CUDA.

4. Jak radzić sobie z wieloma zainstalowanymi wersjami CUDA

W środowisku Ubuntu można zainstalować wiele wersji CUDA. Może to jednak prowadzić do niejasności, która wersja jest aktualnie aktywna.
W takich przypadkach należy przełączyć się na odpowiednią wersję.

Metoda 1: Przełączanie przy użyciu update-alternatives

W Ubuntu możesz użyć update-alternatives, aby przełączać wersje CUDA.

Sprawdź bieżące ustawienia

update-alternatives --query cuda

Przełącz wersję CUDA

sudo update-alternatives --config cuda

Przykładowy wynik

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:

Najważniejsze informacje

• Wykonanie update-alternatives --config cuda wyświetli listę dostępnych wersji CUDA.
• Możesz wybrać żądaną wersję CUDA, wpisując odpowiadający jej numer.
• Dostępne są tryby auto i manual; wybierz manual, jeśli chcesz ręcznie przełączać wersje.

Metoda 2: Ręczne ustawienie linku symbolicznego

Możesz również przełączać wersje CUDA, modyfikując link symboliczny.

Sprawdź istniejący link symboliczny

ls -l /usr/local/cuda

Przykładowy wynik

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

Zmień wersję CUDA

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

Zweryfikuj zmianę

ls -l /usr/local/cuda

Najważniejsze informacje

• /usr/local/cuda jest domyślną ścieżką CUDA. Zmiana tego linku przełącza wersję CUDA.
• Korzystając z polecenia ln -s, możesz łatwo zmienić wersję CUDA bez modyfikacji konfiguracji systemowych.

Dzięki tym metodom możesz efektywnie zarządzać wieloma wersjami CUDA i mieć pewność, że używasz właściwej wersji dla swoich potrzeb.

5. Najczęściej zadawane pytania (FAQ)

Oto kilka typowych pytań związanych ze sprawdzaniem wersji CUDA. Jeśli napotkasz problemy, skorzystaj z poniższych rozwiązań.

P1: Nie znaleziono polecenia nvcc -V!

Jeśli polecenie nvcc nie zostanie znalezione, CUDA może nie być poprawnie zainstalowane lub jego ścieżka nie jest ustawiona.

Rozwiązanie 1: Sprawdź, czy CUDA jest zainstalowane

ls /usr/local/cuda/

Rozwiązanie 2: Dodaj nvcc do ścieżki

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

Po wykonaniu tych poleceń spróbuj ponownie uruchomić nvcc -V, aby sprawdzić, czy wersja jest wyświetlana poprawnie.

P2: Dlaczego wersja CUDA wyświetlana przez nvidia-smi jest inna?

Wersja CUDA wyświetlana przez nvidia-smi reprezentuje maksymalną wersję CUDA obsługiwaną przez sterownik NVIDIA, a niekoniecznie zainstalowaną wersję zestawu narzędzi CUDA.

Jak sprawdzić:

nvidia-smi

Przykładowy wynik:

CUDA Version: 12.1

Aby sprawdzić rzeczywistą zainstalowaną wersję CUDA, użyj nvcc -V lub sprawdź plik version.txt.

P3: Jak sprawdzić kompatybilność CUDA i cuDNN?

Najlepszym sposobem sprawdzenia kompatybilności między CUDA a cuDNN jest odwołanie się do oficjalnej macierzy wsparcia NVIDIA.

Oficjalna dokumentacja:

NVIDIA cuDNN Support Matrix

Dodatkowo możesz sprawdzić zainstalowane wersje, używając następujących poleceń:

Sprawdź wersję CUDA

nvcc -V

Sprawdź wersję cuDNN

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

Poprzez prawidłowe zarządzanie środowiskiem możesz uniknąć problemów z kompatybilnością CUDA i cuDNN.

6. Podsumowanie

W tym artykule wyjaśniliśmy, jak sprawdzić wersję CUDA w środowisku Ubuntu.
Przejrzyjmy kluczowe punkty.

Sposoby sprawdzenia wersji CUDA

Sposoby sprawdzenia wersji cuDNN

Jak przełączać wersje CUDA

Najważniejsze wnioski

• Ważne jest prawidłowe określenie wersji CUDA
• Zapewnij kompatybilność między CUDA a cuDNN
• Jeśli używasz wielu wersji CUDA, zrozum, jak przełączać się między nimi

Poprzez właściwe zarządzanie środowiskiem możesz maksymalnie wykorzystać zalety CUDA.
Mamy nadzieję, że ten artykuł pomoże Ci sprawdzić wersję CUDA w Twoim środowisku Ubuntu.

Powiązane artykuły