एनवीडिया जीपीयू-सक्षम कंटेनर [उबंटू 18.04] का उपयोग करने के लिए डॉकर को कॉन्फ़िगर करें

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

संस्करण 19.03 से शुरू होकर, डॉकर मूल रूप से 'एनवीडिया जीपीयू त्वरित कंटेनर' का समर्थन करता है।यह कार्यक्षमता NVIDIA कंटेनर टूलकिट का उपयोग करके उपलब्ध है।यहां हम इसके लिए जरूरी शुरुआती सेटिंग्स करेंगे।

环境
ओएस: उबंटू 18.04 एलटीएस
जीपीयू: जीटीएक्स1080

चरण XNUMX: पूर्वापेक्षाएँ जाँचें

डॉकर के "जीपीयू के साथ कंटेनर त्वरण" का उपयोग करने से पहले, आइए पर्यावरण की जांच करें।जीपीयू कंटेनरों का उपयोग उन परिवेशों में नहीं किया जा सकता है जो बहुत पुराने हैं।

• GNU/Linux x86_64 (कर्नेल संस्करण > 3.10) ← Ubuntu 18.04 ठीक है
• आर्किटेक्चर के साथ NVIDIA GPU > Fermi (2.1) ← यदि GPU डेस्कटॉप के लिए है, तो GTX600 या बाद का संस्करण ठीक है
• NVIDIA ड्राइवर संस्करण ≧ 361.93 ← नीचे चरण XNUMX देखें

विवरण के लिए कृपया निम्न पृष्ठ देखें।

चरण XNUMX: डॉकर स्थापित करें

यदि आपने अभी तक डॉकर को स्थापित नहीं किया है, तो इसे स्थापित करने के लिए नीचे दिए गए चरणों का पालन करें।

चरण XNUMX: एनवीडिया जीपीयू ड्राइवर स्थापित करें

एनवीडिया ड्राइवरों को जीपीयू कंटेनर (सीयूडीए कंटेनर) चलाने की आवश्यकता है।

Nvidia ड्राइवर संस्करण आपके द्वारा उपयोग किए जाने वाले CUDA संस्करण के साथ संगत से अधिक होना चाहिए (उदाहरण के लिए टेंसरफ़्लो आदि को आपके द्वारा उपयोग किए जाने वाले संस्करण के आधार पर अलग-अलग सटीक CUDA संस्करणों की आवश्यकता होती है)।

जब तक आप एनवीडिया ड्राइवरों को ठीक से स्थापित करते हैं, आपको अलग से CUDA टूलकिट या NVIDIA कंटेनर टूलकिट स्थापित करने की आवश्यकता नहीं है।

यहां एनवीडिया ड्राइवर्स को स्थापित करने का तरीका बताया गया है।

चरण XNUMX: डॉकटर को जीपीयू कंटेनरों का उपयोग करने के लिए सक्षम करें

अपने टर्मिनल से निम्नलिखित दर्ज करें:
आधिकारिक संदर्भ: https://github.com/NVIDIA/nvidia-docker

# पैकेज रिपॉजिटरी वितरण जोड़ें = $(. -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 पुनरारंभ डॉकर

चेक ऑपरेशन

# नवीनतम आधिकारिक CUDA छवि के साथ परीक्षण docker run --gpus all nvidia/cuda:9.0-base nvidia-smi

अगर सब कुछ ठीक रहा तो आपको कुछ ऐसा देखने को मिलेगा।

अब से, GPU कंटेनर का उपयोग करने के लिए, "-gpus" विकल्प को "docker run" में जोड़ें।

यदि आपने अभी डॉकर स्थापित किया है, तो यह तब तक नहीं चल सकता जब तक आप डॉकर चलाने से पहले सूडो नहीं जोड़ते।
[उबंटू] सुडो कमांड के बिना डॉकर चलाने के लिए सेटिंग्स और नोट्स

ऊपर उबंटू पर डॉकर जीपीयू-सक्षम कंटेनरों का उपयोग करने के लिए आवश्यक प्रारंभिक सेटिंग्स हैं।

सन्दर्भ पुसतक