กำหนดค่า Docker เพื่อใช้คอนเทนเนอร์ที่เปิดใช้งาน Nvidia GPU [Ubuntu 18.04]

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

ตั้งแต่เวอร์ชัน 19.03 เป็นต้นมา Docker รองรับ "การเร่งคอนเทนเนอร์ด้วย Nvidia GPUs" โดยกำเนิดฟังก์ชันนี้มีให้ใช้งานโดยใช้ NVIDIA Container Toolkitที่นี่เราจะทำการตั้งค่าเริ่มต้นที่จำเป็นสำหรับมัน

สิ่งแวดล้อม
ระบบปฏิบัติการ: Ubuntu 18.04 LTS
จีพียู: GTX1080

ขั้นตอนที่ XNUMX: ตรวจสอบข้อกำหนดเบื้องต้น

ก่อนที่จะใช้ "การเร่งคอนเทนเนอร์ด้วย GPU" ของ Docker ให้ตรวจสอบสภาพแวดล้อมก่อนไม่สามารถใช้คอนเทนเนอร์ GPU ในสภาพแวดล้อมที่เก่าเกินไป

• GNU/Linux x86_64 (เวอร์ชันเคอร์เนล > 3.10) ← Ubuntu 18.04 ใช้ได้
• NVIDIA GPU พร้อมสถาปัตยกรรม > Fermi (2.1) ← หากเป็น GPU สำหรับเดสก์ท็อป GTX600 หรือใหม่กว่าก็ใช้ได้
• ไดรเวอร์ NVIDIA เวอร์ชัน ≧ 361.93 ← ดูขั้นตอนที่ XNUMX ด้านล่าง

โปรดดูหน้าต่อไปนี้สำหรับรายละเอียด

ขั้นตอนที่ XNUMX: ติดตั้งนักเทียบท่า

หากคุณยังไม่ได้ติดตั้ง Docker ให้ทำตามขั้นตอนด้านล่างเพื่อติดตั้ง

ขั้นตอนที่ XNUMX: ติดตั้งไดรเวอร์ GPU ของ Nvidia

ต้องใช้ไดรเวอร์ของ Nvidia เพื่อเรียกใช้คอนเทนเนอร์ GPU (คอนเทนเนอร์ CUDA)

เวอร์ชันไดรเวอร์ของ Nvidia ควรเข้ากันได้กับเวอร์ชัน CUDA ที่คุณต้องการใช้มากกว่า (เช่น tensorflow เป็นต้น ต้องใช้เวอร์ชัน CUDA ที่แตกต่างกันขึ้นอยู่กับเวอร์ชันที่คุณใช้)

ตราบใดที่คุณติดตั้งไดรเวอร์ Nvidia อย่างถูกต้อง คุณไม่จำเป็นต้องติดตั้งชุดเครื่องมือ CUDA หรือชุดเครื่องมือ NVIDIA Container Toolkit แยกต่างหาก

ต่อไปนี้คือวิธีการติดตั้งไดรเวอร์ของ Nvidia

ขั้นตอนที่ XNUMX: เปิดใช้งาน Docker เพื่อใช้คอนเทนเนอร์ GPU

ป้อนข้อมูลต่อไปนี้จากเทอร์มินัลของคุณ:
ข้อมูลอ้างอิงอย่างเป็นทางการ: 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 รีสตาร์ทนักเทียบท่า

ตรวจสอบการทำงาน

# ทดสอบด้วยการรัน CUDA image docker อย่างเป็นทางการล่าสุด --gpus all nvidia/cuda:9.0-base nvidia-smi

ถ้าทุกอย่างเป็นไปด้วยดี คุณจะเห็นสิ่งนี้

จากนี้ไป หากต้องการใช้คอนเทนเนอร์ GPU ให้เพิ่มตัวเลือก "-gpus" ไปที่ "docker run"

หากคุณเพิ่งติดตั้ง docker อาจไม่ทำงานจนกว่าคุณจะเพิ่ม sudo ก่อน docker run
[Ubuntu] การตั้งค่าและหมายเหตุสำหรับการเรียกใช้นักเทียบท่าโดยไม่ต้องใช้คำสั่ง sudo

ด้านบนคือการตั้งค่าเริ่มต้นที่จำเป็นสำหรับการใช้คอนเทนเนอร์ที่เปิดใช้งาน Docker GPU บน Ubuntu

หนังสืออ้างอิง