Cara Menginstal CUDA di Ubuntu 20.04

CUDA adalah platform komputasi paralel dan model pemrograman yang memberikan pengalaman pengguna yang luar biasa saat memanfaatkan GPU untuk komputasi tujuan umum sehari-hari. CUDA adalah singkatan dari C menghitung U terpadu D perangkat A arsitektur. Itu dibuat oleh NVIDIA.

CUDA terdiri dari CUDA toolkit ( compiler, profile, dan debugger ), driver perangkat lunak, dan CUDA SDK. Lapisan perangkat lunak memberikan akses langsung ke set instruksi virtual GPU dan elemen komputasi paralel. Untuk penelitian pembelajaran mendalam dan pengembang kerangka kerja, gunakan cuDNN untuk akselerasi GPU performa tinggi. Ini adalah perpustakaan primitif untuk jaringan saraf dalam.

Dalam panduan ini, kita mempelajari cara menginstal CUDA dan CuDNN di Ubuntu 20.04 . Ada dua cara untuk melakukan ini. Anda dapat menginstal CUDA dari Repositori Ubuntu - yang merupakan pendekatan termudah - atau menginstal dari repositori CUDA yang menyediakan CUDA versi terbaru.

Prasyarat:

1. Minimal 20 GB ruang disk kosong
2. RAM 2 GB yang disarankan
3. Harus menginstal driver grafis NVIDIA. Driver yang disertakan dengan CUDA mungkin bukan versi terbaru.

Metode 1:Instal CUDA dari repositori Ubuntu

Mulailah dengan memperbarui daftar paket pada instance Ubuntu 20.04 Anda.

$ sudo apt update

Selanjutnya, instal toolkit CUDA menggunakan manajer paket APT sebagai berikut. Perhatikan bahwa pemasangannya intensif, dan karenanya pastikan Anda memiliki koneksi internet yang cepat dan stabil serta ruang disk minimal 10G.

$ sudo apt install nvidia-cuda-toolkit

Setelah instalasi selesai, kita perlu menambahkan CUDA ke PATH untuk memberi tahu shell tentang lokasi CUDA. Untuk melakukan ini, kami akan menentukan PATH di file .bashrc. Jadi, buka file menggunakan editor teks baris perintah Anda.

$ sudo vim ~/.bashrc

Di akhir file, salin dan tempel baris kode berikut.

# 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

Selanjutnya, muat ulang perubahan pada file .bashrc.

$ source ~/.bashrc

Sekarang periksa dan konfirmasi versi CUDA yang diinstal menggunakan perintah berikut.

$ nvcc --version

Metode 2:Instal CUDA dari repositori CUDA

Seperti disebutkan sebelumnya, repositori CUDA menyediakan CUDA versi terbaru. Jadi, jika Anda ingin menginstal versi terbaru, ikuti petunjuk yang diuraikan di sini.

Pertama, unduh pin repositori CUDA:

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

Selanjutnya, pindahkan pin ke /etc/preferences.d direktori dan ganti namanya menjadi cuda-repository-pin-600 .

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

Setelah itu, unduh file CUDA Debian. Ketahuilah bahwa file tersebut cukup besar - 2.3G tepatnya pada saat menulis panduan ini. Pastikan Anda memiliki koneksi internet yang baik dan ruang disk yang cukup.

$ 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

Setelah diunduh, gunakan dpkg utilitas untuk mengeksekusi file.

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

Selanjutnya, instal kunci CUDA GPG sebagai berikut:

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

Terakhir, perbarui daftar paket dan instal CUDA menggunakan manajer paket APT.

$ sudo apt update

$ sudo apt install cuda -y

Instal CuDNN

CuDNN tidak datang dengan Cuda. Untuk mengunduh CuDNN, Anda harus mendaftar untuk menjadi anggota Program Pengembang NVIDIA yang gratis.

Setelah mendaftar di Program Pengembang NVIDIA, Anda dapat mengunduh cuDNN versi terbaru. Di sini saya mengunduh Pustaka cuDNN untuk Linux.

Pertama, ekstrak paket cuDNN:

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

Kemudian salin file berikut ke direktori 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*

Kesimpulan

Dalam tutorial ini, kita belajar cara menginstal CUDA dan cuDNN di Ubuntu 20.04. Terima kasih telah membaca, silakan tinggalkan umpan balik dan saran Anda.