Konfigurasikan Docker untuk menggunakan wadah yang mendukung GPU Nvidia [Ubuntu 18.04]

2019 11 年 月 日 182020 1 年 月 日 15

Dimulai dengan versi 19.03, Docker secara bawaan mendukung "Akselerasi kontainer dengan GPU Nvidia".Fungsionalitas ini tersedia menggunakan NVIDIA Container Toolkit.Di sini kami akan membuat pengaturan awal yang diperlukan untuk itu.

Lingkungan
Sistem Operasi: Ubuntu 18.04 LTS
GPU: GTX1080

Langkah XNUMX: Periksa prasyarat

Sebelum menggunakan "Akselerasi kontainer dengan GPU" Docker, mari kita periksa lingkungannya.Kontainer GPU tidak dapat digunakan di lingkungan yang terlalu tua.

• GNU/Linux x86_64 (versi kernel > 3.10) ← Ubuntu 18.04 OK
• GPU NVIDIA dengan Arsitektur > Fermi (2.1) ← Jika GPU untuk desktop, GTX600 atau yang lebih baru tidak apa-apa
• Versi driver NVIDIA ≧ 361.93 ← Lihat langkah XNUMX di bawah

Silakan lihat halaman berikut untuk detailnya.

Langkah XNUMX: Instal Docker

Jika Anda belum menginstal Docker, ikuti langkah-langkah di bawah ini untuk menginstalnya.

Langkah XNUMX: Instal Driver GPU Nvidia

Driver Nvidia diperlukan untuk menjalankan wadah GPU (wadah CUDA).

Versi driver Nvidia harus lebih dari kompatibel dengan versi CUDA yang ingin Anda gunakan (misalnya tensorflow dll. memerlukan versi CUDA yang berbeda tergantung pada versi mana yang Anda gunakan).

Selama Anda menginstal driver Nvidia dengan benar, Anda tidak perlu menginstal CUDA toolkit atau NVIDIA Container Toolkit secara terpisah.

Berikut cara menginstal Driver Nvidia.

Langkah XNUMX: Aktifkan Docker untuk menggunakan wadah GPU

Masukkan yang berikut dari terminal Anda:
Referensi resmi: https://github.com/NVIDIA/nvidia-docker

# Tambahkan paket repositori distribution=$(. /etc/os-release;echo $ID$VERSION_ID) curl -s -L https://nvidia.github.io/nvidia-docker/gpgkey | sudo apt-key add - curl -s -L https://nvidia.github.io/nvidia-docker/$distribution/nvidia-docker.list | sudo tee /etc/apt/sources.list.d/nvidia-docker.list sudo apt-get update && sudo apt-get install -y nvidia-container-toolkit sudo systemctl restart docker

operasi pemeriksaan

# uji dengan menjalankan docker gambar CUDA resmi terbaru --gpus semua nvidia/cuda:9.0-base nvidia-smi

Jika semuanya berjalan dengan baik, Anda akan melihat sesuatu seperti ini.

Mulai sekarang, untuk menggunakan wadah GPU, tambahkan opsi "-gpus" ke "docker run".

Jika Anda baru saja menginstal buruh pelabuhan, itu mungkin tidak berjalan kecuali Anda menambahkan sudo sebelum buruh pelabuhan dijalankan.
[Ubuntu] Pengaturan dan catatan untuk menjalankan buruh pelabuhan tanpa perintah sudo

Di atas adalah pengaturan awal yang diperlukan untuk menggunakan kontainer berkemampuan GPU Docker di Ubuntu.

buku referensi