Panduan Lengkap untuk Memasang CUDA pada Ubuntu [Mesra Pemula]

1. Pengenalan

CUDA (Compute Unified Device Architecture) ialah platform pengkomputeran selari dan API yang disediakan oleh NVIDIA, membolehkan pengiraan berkelajuan tinggi menggunakan GPU.
Ia banyak digunakan dalam pelbagai bidang, termasuk pembelajaran mesin, pembelajaran mendalam, dan pengkomputeran saintifik.
Panduan ini memberikan penjelasan terperinci tentang cara memasang CUDA pada sistem Ubuntu.

2. Prasyarat

2.1 Memeriksa Keserasian GPU

Mula-mula, periksa sama ada GPU NVIDIA anda menyokong CUDA.
Jalankan perintah berikut dalam terminal:

lspci | grep -i nvidia

Jika sistem anda mengenali per NVIDIA dalam output, GPU anda telah dikesan.
Untuk senarai lengkap GPU yang disokong, rujuk laman web rasmi NVIDIA.

2.2 Memeriksa Versi Ubuntu Anda

CUDA disokong pada versi Ubuntu tertentu.
Periksa versi Ubuntu semasa anda dengan menjalankan perintah berikut:

lsb_release -a

Secara amnya, versi LTS (Long Term Support) Ubuntu disarankan.
Untuk butiran keserasian terkini, rujuk dokumentasi rasmi NVIDIA.

2.3 Memeriksa Pemasangan GCC

Pengkompil GCC diperlukan untuk memasang CUDA.
Sahkan sama ada ia telah dipasang dengan perintah berikut:

gcc --version

Jika GCC belum dipasang, pasang ia dengan menjalankan:

sudo apt install build-essential

3. Memasang Pemacu NVIDIA

3.1 Mengeluarkan Pemacu Sedia Ada

Jika pemacu NVIDIA lama telah dipasang, keluarkan ia untuk mengelakkan konflik.
Jalankan perintah berikut:

sudo apt-get --purge remove '*nvidia*'
sudo apt-get autoremove

3.2 Memilih dan Memasang Pemacu yang Sesuaiak laman web rasmi NVIDIA untuk mencari pemacu yang betul bagi GPU anda, kemudian ikuti langkah-langkah berikut untuk pemasangan:

1. Tambah Repositori: Jalankan perintah berikut dalam terminal untuk menambah repositori pemacu NVIDIA.

   sudo add-apt-repository ppa:graphics-drivers/ppa
   sudo apt-get update

2. Semak Pemacu Disyorkan: Gunakan perintah ini untuk melihat pem yang disyorkan.

   ubuntu-drivers devices

Cari pemacu yang ditandakan sebagai “disyorkan” dalam output.

3. ang Pemacu: Pasang pemacu yang disyorkan dengan menyatakan versinya.

   sudo apt install nvidia-driver-<recommended version>

4. Mula Semula Sistem: Selepas pemasangan, mulakan semula sistem anda.

   sudo reboot

4. Memasang CUDA Toolkit

4.1 Memilih Versi CUDA

Lawati halaman muat turun rasmi CUDA untuk mencari versi CUDA yang serasi dengan GPU dan versi Ubuntu anda.
Jika menggunakan versi terkini, pastikan keserasian dengan perisian dan perpustakaan anda.

4.2 Menambah Repositori dan Memasang CUDA

Ikuti langkah-langkah berikut untuk memasang CUDA Toolkit.

1. Tambah Repositori: Tambah repositori NVIDIA (contoh untuk Ubuntu 20.04).

   wget https://developer.download.nvidia.com/compute/cuda/repos/ubuntu2004/x86_64/cuda-ubuntu2004.pin
   sudo mv cuda-ubuntu2004.pin /etc/apt/preferences.d/cuda-repository-pin-600

2. Tambah Kunci Repositori: Dapatkan dan pasang kunci repositori.

   sudo apt-key adv --fetch-keys https://developer.download.nvidia.com/compute/cuda/repos/ubuntu2004/x86_64/7fa2af80.pub

3. Pasang Pakej CUDA: Pasang CUDA Toolkit.

   sudo apt update
   sudo apt install cuda

4. Sahkan Pemasangan: Periksa sama ada CUDA dipasang dengan betul.

   nvcc --version

5. Memasang cuDNN

5.1 Memuat Turun cuDNN

cuDNN (CUDA Deep Neural Network library) ialah perpustakaan dipercepatkan GPU NVIDIA untuk pembelajaran mendalam.
Untuk memasang cuDNN, ikuti langkah-langkah berikut:

1. Pergi ke halaman muat turun NVIDIA cuDNN dan log masuk.
2. Pilih versi cuDNN yang sesuai dengan versi CUDA anda.
3. Muat turun pakej cuDNN untuk Ubuntu.

5.2 Memasang cuDNN

Selepas memuat turun, pasang cuDNN menggunakan perintah berikut:

tar -xvf cudnn-*.tar.xz
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*

Selepas pemasangan, sahkan bahawa cuDNN dipasang dengan betul dengan memeriksa versinya:

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

6. Menetapkan Pembolehubah Persekitaran

Selepas memasang CUDA dan cuDNN, kemas kini pembolehubah persekitaran untuk memastikan ia dikenali dengan betul oleh sistem.

6.1 Mengemas kini .bashrc

Edit fail .bashrc untuk menambah laluan CUDA:

echo 'export PATH=/usr/local/cuda/bin:$PATH' >> ~/.bashrc
echo 'export LD_LIBRARY_PATH=/usr/local/cuda/lib64:$LD_LIBRARY_PATH' >> ~/.bashrc
source ~/.bashrc

6.2 Mengesahkan Pemasangan CUDA

Untuk mengesahkan bahawa CUDA dipasang dan dikonfigurasikan dengan betul, jalankan perintah berikut:

nvcc --version

Ini sepatutnya memaparkan versi CUDA yang dipasang pada sistem anda.

7. Menjalankan Program Ujian

Untuk memeriksa sama ada CUDA berfungsi dengan betul, susun dan jalankan program ujian ringkas.

#include <stdio.h>
int main() {
    printf("CUDA setup is complete!n");
    return 0;
}

Susun dan jalankannya menggunakan:

gcc test.c -o test
./test

8. Penyelesaian Masalah

8.1 Isu Umum dan Penyelesaian

• CUDA tidak dikenali: Pastikan pembolehubah persekitaran ditetapkan dengan betul dengan menjalankan echo $PATH dan echo $LD_LIBRARY_PATH .
• Masalah pemacu: Jika pemacu NVIDIA tidak berfungsi, cuba pasang semula dengan menggunakan langkah-langkah dalam bahagian 3.
• Versi cuDNN tidak sepadan: Semak bahawa versi cuDNN anda serasi dengan versi CUDA yang dipasang.

9. Kesimpulan

Dengan mengikuti panduan ini, anda seharusnya kini mempunyai persekitaran CUDA yang berfungsi sepenuhnya yang dipasang pada Ubuntu.
Dengan CUDA dan cuDNN dipasang, anda boleh mula menggunakan pemecutan GPU untuk pembelajaran mendalam, pengiraan saintifik, dan aplikasi berprestasi tinggi yang lain.

Jika anda menghadapi sebarang masalah, rujuk dokumentasi rasmi NVIDIA atau forum komuniti untuk sokongan tambahan.