Cara Menggunakan GPU NVIDIA di Docker Containers di Ubuntu 22.04 LTS

Cara Menggunakan Gpu Nvidia Di Docker Containers Di Ubuntu 22 04 Lts



Anda dapat melewati GPU NVIDIA Anda dalam wadah Docker dan menjalankan program CUDA pada GPU NVIDIA Anda dari wadah Docker ini. Ini adalah fitur yang sangat berguna untuk mempelajari AI (Kecerdasan Buatan). Mampu menjalankan kode AI (yaitu Tensorflow) pada kontainer Docker akan menghemat banyak waktu Anda. Anda juga dapat mencoba kode AI Anda pada versi CUDA yang berbeda tanpa perlu mengubah versi CUDA di komputer Anda.

Mengutak-atik driver NVIDIA dan versi CUDA di komputer Anda dapat menyebabkan masalah seperti driver NVIDIA tidak berfungsi atau meninggalkan Anda dengan layar kematian hitam/biru. Jadi, selalu merupakan ide bagus untuk membiarkan sistem Anda apa adanya dan membuat perubahan di lingkungan yang terisolasi seperti kontainer Docker. Ini menjaga komputer utama Anda bersih (dari alat pengembangan yang tidak perlu).







Pada artikel ini, saya akan menunjukkan kepada Anda cara mengatur Docker CE dan NVIDIA docker di Ubuntu 22.04 LTS sehingga Anda dapat mengakses GPU NVIDIA komputer Anda dari wadah Docker dan menjalankan program CUDA pada GPU NVIDIA Anda.



Daftar isi:

  1. Memeriksa apakah Driver GPU NVIDIA Resmi Dipasang di Ubuntu 22.04
  2. Menginstal Ketergantungan Docker CE di Ubuntu 22.04
  3. Menginstal Kunci GPG Docker CE di Ubuntu 22.04
  4. Menginstal Repositori Docker CE di Ubuntu 22.04
  5. Menginstal Docker CE di Ubuntu 22.04
  6. Menambahkan Pengguna Masuk Ubuntu 22.04 LTS ke Grup buruh pelabuhan
  7. Memeriksa apakah Docker CE Dipasang dengan Benar di Ubuntu 22.04
  8. Menginstal NVIDIA Container Toolkit GPG Keys di Ubuntu 22.04
  9. Menginstal Repositori NVIDIA Container Toolkit di Ubuntu 22.04
  10. Menginstal Driver nvidia-docker di Ubuntu 22.04
  11. Memeriksa apakah NVIDIA GPU Dapat Diakses dari Docker Containers di Ubuntu 22.04
  12. Kesimpulan
  13. Referensi

Memeriksa apakah Driver GPU NVIDIA Resmi Dipasang di Ubuntu 22.04 LTS:

Sebelum memulai, jalankan perintah berikut untuk memverifikasi bahwa Anda telah menginstal GPU NVIDIA di komputer Anda.



$ lspci | egrep -Saya 'vga|3d|tampilan'





Dalam hal ini, saya memasang GPU NVIDIA GTX 1050 Ti di komputer saya. Kemungkinan besar Anda akan menginstal GPU NVIDIA yang berbeda di komputer Anda.


Juga, pastikan Anda memiliki driver NVIDIA resmi yang diinstal pada sistem operasi Ubuntu 22.04 Anda dengan perintah berikut:



$ lsmod | pegangan nvidia

Jika driver NVIDIA resmi diinstal pada sistem operasi Ubuntu 22.04 Anda, Anda akan melihat keluaran berikut.


Juga, periksa apakah driver GPU NVIDIA resmi berfungsi dengan perintah berikut:

$ nvidia-smi

Jika driver GPU NVIDIA resmi berfungsi, Anda akan melihat output berikut. Seperti yang Anda lihat, saya telah menginstal NVIDIA GPU Driver resmi versi 525.78.01 di mesin Ubuntu 22.04 saya.


Jika Anda tidak menginstal driver GPU NVIDIA resmi pada mesin Ubuntu 22.04 Anda dan Anda memerlukan bantuan untuk itu, silakan periksa artikelnya .

Menginstal Ketergantungan Docker CE di Ubuntu 22.04 LTS:

Sebelum Anda dapat menginstal Docker CE di Ubuntu 22.04, Anda harus menginstal paket dependensi Docker CE yang diperlukan di Ubuntu 22.04.

Pertama, perbarui cache repositori paket APT dengan perintah berikut:

$ sudo pembaruan tepat

Cache repositori paket APT harus diperbarui.


Untuk menginstal paket ketergantungan Docker CE yang diperlukan, jalankan perintah berikut:

$ sudo apt-get instal ca-sertifikat curl gnupg lsb-rilis

Untuk mengonfirmasi pemasangan, tekan DAN lalu tekan .


Paket dependensi Docker CE yang diperlukan harus diinstal.

Menginstal Kunci GPG Docker CE di Ubuntu 22.04 LTS:

Di bagian ini, saya akan menunjukkan cara menginstal kunci GPG dari repositori paket Docker CE di Ubuntu 22.04.

Pertama, buat folder baru /etc/apt/keyrings dengan perintah berikut:

$ sudo mkdir -P / dll. / tepat / gantungan kunci

Untuk menginstal kunci GPG dari repositori paket Docker CE di Ubuntu 22.04, jalankan perintah berikut:

$ keriting -fsSL https: // download.docker.com / linux / ubuntu / gpg | sudo gpg --sayang -HAI / dll. / tepat / gantungan kunci / docker.gpg

Menginstal Repositori Docker CE di Ubuntu 22.04 LTS:

Untuk menginstal repositori paket Docker CE di Ubuntu 22.04, jalankan perintah berikut:

$ gema 'deb [lengkungan= $(dpkg --print-arsitektur) ditandatangani-oleh=/etc/apt/keyrings/docker.gpg] https://download.docker.com/linux/ubuntu $(lsb_release -cs) stabil' | sudo tee / dll. / tepat / sources.list.d / docker.list > / dev / batal

Perbarui cache repositori paket APT agar perubahan diterapkan.

$ sudo pembaruan tepat


Menginstal Docker CE di Ubuntu 22.04 LTS:

Untuk menginstal Docker CE versi terbaru di Ubuntu 22.04, jalankan perintah berikut:

$ sudo apt-get instal docker-ce docker-ce-cli containerd.io docker-compose-plugin

Untuk mengonfirmasi pemasangan, tekan DAN lalu tekan .


Docker CE dan paket dependensi yang diperlukan sedang diunduh. Butuh beberapa saat untuk menyelesaikannya.


Docker CE dan paket dependensi yang diperlukan sedang diinstal. Butuh beberapa saat untuk menyelesaikannya.


Docker CE dan paket dependensi yang diperlukan harus diinstal pada saat ini.

Menambahkan Pengguna Masuk Ubuntu 22.04 LTS ke Grup buruh pelabuhan:

Untuk membuat wadah Docker dan mengelolanya tanpa menggunakan sudo atau login sebagai pengguna root, Anda harus menambahkan pengguna login Anda ke buruh pelabuhan kelompok.

Untuk menambahkan pengguna login Ubuntu 22.04 Anda ke buruh pelabuhan grup, jalankan perintah berikut:

$ sudo usermod -aG buruh pelabuhan $ ( siapa saya )

Agar perubahan diterapkan, mulai ulang komputer Anda dengan perintah berikut:

$ sudo menyalakan ulang

Memeriksa apakah Docker CE Dipasang dengan Benar di Ubuntu 22.04 LTS:

Setelah komputer Anda mulai, jalankan perintah berikut untuk memverifikasi bahwa Anda dapat mengakses Docker tanpa hak superuser.

$ versi buruh pelabuhan

Jika semuanya berfungsi, Anda akan melihat output berikut.

Seperti yang Anda lihat, saya menjalankan Docker versi 20.10.23 – versi terbaru Docker CE pada saat penulisan ini.

Menginstal Kunci GPG NVIDIA Container Toolkit di Ubuntu 22.04:

Pada bagian ini, saya akan menunjukkan cara menginstal kunci GPG dari repositori paket NVIDIA Container Toolkit di Ubuntu 22.04.

Untuk menginstal kunci GPG dari repositori paket NVIDIA Container Toolkit di Ubuntu 22.04, jalankan perintah berikut:

$ keriting -fsSL https: // nvidia.github.io / libnvidia-container / gpgkey | sudo gpg --sayang -HAI / usr / membagikan / gantungan kunci / nvidia-container-toolkit-keyring.gpg

Menginstal Repositori NVIDIA Container Toolkit di Ubuntu 22.04 LTS:

Di bagian ini, saya akan menunjukkan cara menginstal repositori paket NVIDIA Container Toolkit di Ubuntu 22.04.

Pertama, buat file sumber APT baru nvidia-container-toolkit.list dalam /etc/apt/sources.list.d/ direktori sebagai berikut:

$ sudo nano / dll. / tepat / sources.list.d / nvidia-container-toolkit.list

Dalam nvidia-container-toolkit.list file, tambahkan baris berikut dan tekan + X diikuti oleh DAN Dan untuk menyimpan file.

itu [ ditandatangani-oleh= / usr / membagikan / gantungan kunci / nvidia-container-toolkit-keyring.gpg ] https: // nvidia.github.io / libnvidia-container / stabil / gratis22.04 / $ ( LENGKUNGAN ) /



Jika Anda menggunakan Ubuntu 20.04 LTS, maka yang perlu Anda lakukan hanyalah menambahkan baris berikut di nvidia-container-toolkit.list file sebagai gantinya dan yang lainnya harus berfungsi tidak berubah.

itu [ ditandatangani-oleh= / usr / membagikan / gantungan kunci / nvidia-container-toolkit-keyring.gpg ] https: // nvidia.github.io / libnvidia-container / stabil / gratis20.04 / $ ( LENGKUNGAN ) /


Jika Anda menggunakan Ubuntu 18.04 LTS, maka yang perlu Anda lakukan hanyalah menambahkan baris berikut di nvidia-container-toolkit.list file sebagai gantinya dan yang lainnya harus berfungsi tidak berubah.

itu [ ditandatangani-oleh= / usr / membagikan / gantungan kunci / nvidia-container-toolkit-keyring.gpg ] https: // nvidia.github.io / libnvidia-container / stabil / gratis18.04 / $ ( LENGKUNGAN ) /


Agar perubahan diterapkan, perbarui cache repositori paket APT dengan perintah berikut:

$ sudo pembaruan tepat


Menginstal Driver nvidia-docker di Ubuntu 22.04 LTS:

Untuk menginstal driver NVIDIA Docker di Ubuntu 22.04, jalankan perintah berikut:

$ sudo tepat Install nvidia-docker2

Untuk mengonfirmasi pemasangan, tekan DAN lalu tekan .


Driver buruh pelabuhan NVIDIA harus diinstal.


Agar perubahan diterapkan, mulai ulang komputer Anda dengan perintah berikut:

$ sudo menyalakan ulang

Memeriksa apakah NVIDIA GPU Dapat Diakses dari Docker Containers di Ubuntu 22.04 LTS:

Pada bagian ini, saya akan menunjukkan cara membuat sebuah Wadah Docker NVIDIA CUDA dan verifikasi bahwa wadah dapat mengakses GPU NVIDIA dari komputer Anda.

Untuk membuat wadah Docker NVIDIA CUDA 12 berdasarkan Ubuntu 20.04 LTS dan jalankan nvidia-smi perintah di dalamnya setelah dibuat untuk memverifikasi apakah itu dapat mengakses GPU NVIDIA dari komputer Anda, jalankan perintah berikut:

$ menjalankan buruh pelabuhan --rm --gpus semua nvidia / cuda:12.0.0-base-ubuntu20.04 nvidia-smi

Docker menarik nvidia/cuda:12.0.0-base-ubuntu20.04 gambar dari Docker Hub. Butuh beberapa saat untuk menyelesaikannya.


Setelah gambar docker NVIDIA CUDA ditarik dan wadah dibuat, file nvidia-smi perintah akan berjalan di atasnya dan mencetak hasilnya di konsol seperti yang Anda lihat pada tangkapan layar di bawah.

Kontainer Docker menggunakan Driver GPU NVIDIA 525.78.01 [1] dan CUDA versi 12.0 [2] . Jika Anda melihat keluaran serupa, wadah Docker dapat mengakses GPU NVIDIA komputer Anda.


Jika Anda perlu menggunakan CUDA versi lama, periksa .

Kesimpulan:

Pada artikel ini, saya telah menunjukkan kepada Anda cara mengatur repositori paket Docker CE di Ubuntu 22.04. Saya juga telah menunjukkan kepada Anda cara menginstal Docker CE versi terbaru di Ubuntu 22.04. Saya telah menunjukkan kepada Anda cara menginstal repositori paket NVIDIA Container Toolkit di Ubuntu 22.04 serta cara menginstal driver NVIDIA Docker di Ubuntu 22.04. Terakhir, saya telah menunjukkan kepada Anda cara mengakses GPU NVIDIA komputer Anda dari wadah Docker.

Referensi: