Cara Memeriksa Versi CUDA pada Ubuntu [Panduan Perintah Mudah]

目次

1. Pengenalan

CUDA (Compute Unified Device Architecture) ialah platform pengkomputeran selari yang dibangunkan oleh NVIDIA yang menggunakan GPU. Ia banyak digunakan untuk pembelajaran mesin, pembelajaran mendalam, rendering 3D, dan pelbagai tugas pengiraan lain.

Apabila menggunakan CUDA dalam persekitaran Ubuntu, adalah penting untuk memeriksa versi CUDA atas sebab-sebab berikut:

Keserasian Pemandu

CUDA memerlukan versi khusus pemandu NVIDIA untuk berfungsi dengan betul. Jika versi tidak serasi, CUDA mungkin tidak berfungsi dengan baik.

Keserasian Pustaka

Pustaka seperti TensorFlow dan PyTorch memerlukan versi khusus CUDA dan cuDNN. Adalah penting untuk memastikan anda telah memasang versi yang sesuai.

Mencegah Kekeliruan Sistem

Jika pelbagai versi CUDA dipasang pada sistem, anda perlu mengenal pasti versi mana yang aktif dan menukar antara versi mengikut keperluan.

Dalam artikel ini, kami akan memberikan penjelasan jelas tentang cara memeriksa versi CUDA pada Ubuntu.

2. Cara Memeriksa Versi CUDA pada Ubuntu

Dalam persekitaran Ubuntu, anda boleh memeriksa versi CUDA menggunakan kaedah berikut:

Kaedah 1: Periksa dengan Perintah nvidia-smi (Kaedah Paling Mudah)

Pemandu NVIDIA menyertakan alat yang dipanggil nvidia-smi (NVIDIA System Management Interface) yang membolehkan anda memeriksa status GPU anda.

Perintah Pelaksanaan

nvidia-smi

Contoh Output

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

Perkara Penting

  • CUDA Version: 12.1 yang dipaparkan di sini mewakili versi maksimum CUDA yang disokong oleh pemandu NVIDIA.
  • Ini mungkin tidak selalu sepadan dengan versi toolkit CUDA yang dipasang, jadi disarankan untuk memeriksa menggunakan kaedah tambahan.

Kaedah 2: Periksa dengan Perintah nvcc -V (Untuk Pembangun)

Jika CUDA dipasang dengan betul, anda boleh memeriksa versi nvcc (pengkompil CUDA).

Perintah Pelaksanaan

nvcc -V

Contoh Output

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

Perkara Penting

  • Bahagian yang menyatakan release 12.1, V12.1.105 menunjukkan versi toolkit CUDA yang dipasang.
  • Ini mungkin berbeza dengan versi yang dipaparkan oleh nvidia-smi, jadi berhati-hati.

Kaedah 3: Periksa Fail version.txt (Pengesahan Manual)

Jika CUDA dipasang di /usr/local/cuda, maklumat versi direkodkan dalam fail version.txt.

Perintah Pelaksanaan

cat /usr/local/cuda/version.txt

Contoh Output

CUDA Version 12.1.105

Perkara Penting

  • Kaedah ini berguna jika perintah nvcc -V tidak tersedia.
  • Pastikan /usr/local/cuda dipautkan dengan betul kepada versi CUDA yang dikehendaki.

3. Cara Memeriksa Versi cuDNN

cuDNN (CUDA Deep Neural Network) ialah pustaka yang direka untuk pembelajaran mendalam dan digunakan bersama CUDA.
Selain memeriksa versi CUDA, adalah penting juga untuk mengesahkan versi cuDNN.

Kaedah 1: Periksa Fail cudnn_version.h

Versi cuDNN direkodkan dalam fail header cudnn_version.h.

Perintah Pelaksanaan

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

Contoh Output

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

Perkara Penting

  • Output ini mengesahkan bahawa cuDNN 8.9.1 telah dipasang.
  • Menggunakan perintah grep membolehkan anda dengan mudah mendapatkan maklumat versi cuDNN.
  • Oleh kerana cuDNN mesti serasi dengan CUDA, adalah penting untuk mengesahkan kombinasi versi yang betul.

Kaedah 2: Periksa dengan Perintah dpkg (Untuk Linux Berasaskan Debian)

Pada Ubuntu dan distribusi Linux berasaskan Debian lain, anda boleh memeriksa versi cuDNN yang dipasang menggunakan perintah dpkg.

Perintah Pelaksanaan

dpkg -l | grep libcudnn

Output Contoh

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

Poin Penting

  • Bahagian libcudnn8 8.9.1-1+cuda12.1 mengesahkan versi cuDNN yang dipasang (8.9.1).
  • Bahagian cuda12.1 menunjukkan versi CUDA yang serasi (12.1).

Dengan menggunakan kaedah ini, anda dapat memastikan persekitaran CUDA anda dikonfigurasikan dengan betul.

4. Cara Mengendalikan Pelbagai Versi CUDA yang Dipasang

Dalam persekitaran Ubuntu, pelbagai versi CUDA boleh dipasang. Walau bagaimanapun, ini kadang‑kadang boleh menyebabkan kekeliruan mengenai versi mana yang sedang aktif.
Dalam situasi seperti ini, anda perlu menukar kepada versi yang sesuai.

Kaedah 1: Tukar Menggunakan update-alternatives

Di Ubuntu, anda boleh menggunakan update-alternatives untuk menukar versi CUDA.

Semak Tetapan Semasa

update-alternatives --query cuda

Tukar Versi CUDA

sudo update-alternatives --config cuda

Output Contoh

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:

Poin Penting

  • Menjalankan update-alternatives --config cuda akan memaparkan senarai versi CUDA yang tersedia.
  • Anda boleh memilih versi CUDA yang dikehendaki dengan memasukkan nombor yang sepadan.
  • auto mode dan manual mode tersedia; pilih manual mode jika anda ingin menukar versi secara manual.

Kaedah 2: Tetapkan Pautan Simbolik Secara Manual

Anda juga boleh menukar versi CUDA dengan mengubah pautan simbolik.

Semak Pautan Simbolik Sedia Ada

ls -l /usr/local/cuda

Output Contoh

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

Tukar Versi CUDA

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

Sahkan Perubahan

ls -l /usr/local/cuda

Poin Penting

  • /usr/local/cuda berfungsi sebagai laluan CUDA lalai. Menukar pautan ini menukar versi CUDA.
  • Dengan menggunakan perintah ln -s, anda boleh menukar versi CUDA dengan mudah tanpa mengubah konfigurasi sistem secara keseluruhan.

Dengan kaedah-kaedah ini, anda dapat menguruskan pelbagai versi CUDA dengan cekap dan memastikan anda menggunakan versi yang betul mengikut keperluan.

5. Soalan Lazim (FAQ)

Berikut adalah beberapa soalan umum yang berkaitan dengan cara memeriksa versi CUDA. Jika anda menghadapi sebarang masalah, rujuk penyelesaian di bawah.

S1: Perintah nvcc -V Tidak Ditemui!

Jika perintah nvcc tidak ditemui, kemungkinan CUDA tidak dipasang dengan betul, atau laluan (path)nya belum ditetapkan.

Penyelesaian 1: Semak Sama Ada CUDA Dipasang

ls /usr/local/cuda/

Penyelesaian 2: Tambah nvcc ke Laluan

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

Selepas menjalankan perintah-perintah ini, cuba jalankan semula nvcc -V untuk memastikan versi dipaparkan dengan betul.

S2: Mengapa Versi CUDA yang Dipaparkan oleh nvidia-smi Berbeza?

Versi CUDA yang dipaparkan oleh nvidia-smi mewakili versi maksimum CUDA yang disokong oleh pemacu NVIDIA, bukan semestinya versi toolkit CUDA yang dipasang.

Cara Memeriksa:

nvidia-smi

Output Contoh:

CUDA Version: 12.1

Untuk memeriksa versi CUDA yang sebenarnya dipasang, gunakan nvcc -V atau semak fail version.txt.

S3: Bagaimana Memeriksa Keserasian CUDA dan cuDNN?

Cara terbaik untuk memeriksa keserasian antara CUDA dan cuDNN ialah merujuk kepada matriks sokongan rasmi NVIDIA.

Dokumentasi Rasmi:

NVIDIA cuDNN Support Matrix

Selain itu, anda boleh menyemak versi yang dipasang menggunakan arahan berikut:

Semak Versi CUDA

nvcc -V

Semak Versi cuDNN

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

Dengan mengurus persekitaran anda dengan betul, anda dapat mengelakkan masalah keserasian CUDA dan cuDNN.

6. Ringkasan

Dalam artikel ini, kami menerangkan cara menyemak versi CUDA dalam persekitaran Ubuntu.
Mari kita semak poin-poin utama.

Cara-cara Menyemak Versi CUDA

MethodCommandDescription
nvidia-sminvidia-smiShows the CUDA version supported by the NVIDIA driver
nvcc -Vnvcc -VShows the actual installed CUDA toolkit version
version.txtcat /usr/local/cuda/version.txtManually check the CUDA version

Cara-cara Menyemak Versi cuDNN

MethodCommandDescription
cudnn_version.hcat /usr/local/cuda/include/cudnn_version.h | grep CUDNN_MAJOR -A 2Check the version from the header file
dpkg Commanddpkg -l | grep libcudnnCheck the installed cuDNN version

Cara Menukar Versi CUDA

MethodCommandDescription
update-alternativessudo update-alternatives --config cudaSwitch between multiple CUDA versions
Symbolic Linksudo ln -s /usr/local/cuda-XX.X /usr/local/cudaManually change the CUDA version

Intipati Utama

  • Adalah penting untuk mengenal pasti versi CUDA dengan betul
  • Pastikan keserasian antara CUDA dan cuDNN
  • Jika menggunakan pelbagai versi CUDA, fahami cara menukar antara mereka

Dengan mengurus persekitaran anda dengan betul, anda dapat memaksimumkan manfaat CUDA.
Kami berharap artikel ini membantu anda menyemak versi CUDA dalam persekitaran Ubuntu anda.

Artikel Berkaitan