Định cấu hình Docker để sử dụng bộ chứa hỗ trợ GPU Nvidia [Ubuntu 18.04]

Ngày 2019 tháng 11 năm 18Ngày 2020 tháng 1 năm 15

Bắt đầu với phiên bản 19.03, Docker nguyên bản hỗ trợ "Tăng tốc vùng chứa bằng GPU Nvidia".Chức năng này khả dụng khi sử dụng Bộ công cụ chứa NVIDIA.Ở đây chúng tôi sẽ thực hiện các cài đặt ban đầu cần thiết cho nó.

Môi trường
Hệ điều hành: Ubuntu 18.04 LTS
GPU: GTX1080

Bước XNUMX: Kiểm tra điều kiện tiên quyết

Trước khi sử dụng "Tăng tốc vùng chứa bằng GPU" của Docker, hãy kiểm tra môi trường.Không thể sử dụng vùng chứa GPU trong môi trường quá cũ.

• GNU/Linux x86_64 (phiên bản kernel > 3.10) ← Ubuntu 18.04 vẫn ổn
• GPU NVIDIA có Kiến trúc > Fermi (2.1) ← Nếu GPU dành cho máy tính để bàn, GTX600 trở lên là được
• Phiên bản trình điều khiển NVIDIA ≧ 361.93 ← Xem bước XNUMX bên dưới

Vui lòng xem trang sau để biết chi tiết.

Bước XNUMX: Cài đặt Docker

Nếu bạn chưa cài đặt Docker, hãy làm theo các bước bên dưới để cài đặt nó.

Bước XNUMX: Cài đặt trình điều khiển GPU Nvidia

Trình điều khiển Nvidia được yêu cầu để chạy bộ chứa GPU (bộ chứa CUDA).

Phiên bản trình điều khiển Nvidia phải tương thích hơn với phiên bản CUDA mà bạn muốn sử dụng (ví dụ: tensorflow, v.v. yêu cầu các phiên bản CUDA chính xác khác nhau tùy thuộc vào phiên bản bạn sử dụng).

Miễn là bạn cài đặt trình điều khiển Nvidia đúng cách, bạn không cần cài đặt riêng bộ công cụ CUDA hoặc Bộ công cụ chứa NVIDIA.

Đây là cách cài đặt Trình điều khiển Nvidia.

Bước XNUMX: Kích hoạt Docker để sử dụng bộ chứa GPU

Nhập thông tin sau từ thiết bị đầu cuối của bạn:
Tài liệu tham khảo chính thức: https://github.com/NVIDIA/nvidia-docker

# Thêm phân phối kho lưu trữ gói=$(. /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 khởi động lại docker

kiểm tra hoạt động

# thử nghiệm với docker hình ảnh CUDA chính thức mới nhất chạy --gpus all nvidia/cuda:9.0-base nvidia-smi

Nếu mọi việc suôn sẻ, bạn sẽ thấy một cái gì đó như thế này.

Từ bây giờ, để sử dụng bộ chứa GPU, hãy thêm tùy chọn "-gpus" vào "docker run".

Nếu bạn mới cài đặt docker, nó có thể không chạy trừ khi bạn thêm sudo trước khi chạy docker.
[Ubuntu] Cài đặt và lưu ý khi chạy docker không cần lệnh sudo

Trên đây là các cài đặt ban đầu cần thiết để sử dụng bộ chứa hỗ trợ GPU Docker trên Ubuntu.

sách tham khảo