قم بإعداد Docker لاستخدام حاوية Nvidia GPU المتوافقة [Ubuntu 18.04]

2019 11 年 月 日 18

بدءًا من الإصدار 19.03 ، يدعم Docker أصلاً "تسريع الحاويات باستخدام وحدات معالجة الرسومات Nvidia".تتوفر هذه الميزة باستخدام مجموعة أدوات حاوية NVIDIA.هنا ، قم بإجراء الإعدادات الأولية المطلوبة لذلك.

環境
نظام التشغيل: Ubuntu 18.04 LTS
وحدة معالجة الرسومات: GTX1080

الخطوة XNUMX: تحقق من المتطلبات الأساسية

قبل استخدام Docker "تسريع الحاويات بواسطة GPU" ، دعنا نتحقق من البيئة.لا يمكن استخدام حاويات GPU في البيئات القديمة جدًا.

  • GNU / Linux x86_64 (إصدار kernel> 3.10) ← Ubuntu 18.04 حسنًا
  • NVIDIA GPU مع الهندسة المعمارية> Fermi (2.1) ← إذا كانت وحدة معالجة الرسومات لسطح المكتب ، فإن GTX 600 أو أحدث على ما يرام
  • إصدار برنامج تشغيل NVIDIA ≧ 361.93 ← انظر الخطوة XNUMX أدناه

يرجى الاطلاع على الصفحة التالية للحصول على التفاصيل.

الخطوة الأولى: تثبيت Docker

إذا لم تقم بتثبيت Docker بعد ، فاتبع الخطوات أدناه لتثبيته.

الخطوة XNUMX: قم بتثبيت برنامج تشغيل Nvidia GPU

مطلوب برنامج تشغيل Nvidia لتشغيل حاوية GPU (حاوية CUDA).

يجب أن يكون إصدار برنامج تشغيل Nvidia أكثر من إصدار CUDA الذي تريد استخدامه (على سبيل المثال ، يتطلب tensorflow إصدار CUDA بالضبط اعتمادًا على الإصدار الذي تستخدمه).

المراسلات بين إصدار CUDA وإصدار Nvidia Driver
المراسلات بين إصدار CUDA وإصدار Nvidia Driver

طالما قمت بتثبيت برنامج تشغيل Nvidia بشكل صحيح ، فلن تحتاج إلى تثبيت مجموعة أدوات CUDA أو مجموعة أدوات حاوية NVIDIA بشكل منفصل.

تم تلخيص طريقة تثبيت برنامج تشغيل Nvidia أدناه.

الخطوة XNUMX: اجعل حاوية GPU متاحة في Docker

أدخل ما يلي من المحطة.
المرجع الرسمي: https://github.com/NVIDIA/nvidia-docker

# إضافة توزيع مستودع الحزم = $ (. / 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 reset docker

تحقق من العملية

# تم اختباره مع أحدث تشغيل رسمي لـ CUDA image docker run --gpus all nvidia / cuda: 9.0-base nvidia-smi

إذا سارت الأمور على ما يرام ، فسترى هذه الشاشة.

لاستخدام حاوية وحدة معالجة الرسومات من الآن فصاعدًا ، أضف الخيار "–gpus" إلى "تشغيل عامل الإرساء".

إذا قمت للتو بتثبيت docker ، فقد لا يعمل إلا إذا قمت بإضافة sudo قبل تشغيل docker.
[Ubuntu] الإعدادات والاحتياطات لتشغيل عامل الإرساء بدون أمر sudo

هذه هي الإعدادات الأولية المطلوبة لاستخدام حاوية Docker المتوافقة مع GPU على Ubuntu.

كتاب مرجعي