Настройте Docker для использования контейнера, совместимого с графическим процессором Nvidia [Ubuntu 18.04]

2019 год 11 месяц 18 день2020 год 1 месяц 15 день

Начиная с версии 19.03, Docker изначально поддерживает «ускорение контейнеров с помощью графических процессоров Nvidia».Эта функция доступна с помощью NVIDIA Container Toolkit.Здесь выполните необходимые начальные настройки.

環境
ОС: Ubuntu 18.04 LTS
Графический процессор: GTX1080

Шаг XNUMX: проверьте предварительные условия

Прежде чем использовать Docker «ускорение контейнеров с помощью GPU», давайте проверим окружение.Контейнеры GPU нельзя использовать в слишком старых средах.

• GNU / Linux x86_64 (версия ядра> 3.10) ← Ubuntu 18.04, так что хорошо
• Графический процессор NVIDIA с архитектурой> Fermi (2.1) ← Если графический процессор предназначен для настольных ПК, подойдет GTX 600 или новее
• Версия драйвера NVIDIA ≧ 361.93 ← См. Шаг XNUMX ниже

Подробности см. На следующей странице.

Шаг XNUMX. Установите Docker

Если вы еще не установили Docker, выполните следующие действия, чтобы установить его.

Шаг XNUMX. Установите драйвер графического процессора Nvidia.

Драйвер Nvidia необходим для запуска контейнера GPU (контейнера CUDA).

Версия драйвера Nvidia должна быть больше, чем версия CUDA, которую вы хотите использовать (например, для tenorflow требуется именно та версия CUDA, в зависимости от используемой вами версии).

Пока вы правильно устанавливаете драйвер Nvidia, вам не нужно отдельно устанавливать CUDA toolkit или NVIDIA Container Toolkit.

Метод установки драйвера 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 restart docker

Проверить работу

# Протестировано с последним официальным докером изображений CUDA run --gpus all nvidia / cuda: 9.0-base nvidia-smi

Если все пойдет хорошо, вы увидите этот дисплей.

Чтобы с этого момента использовать контейнер GPU, добавьте параметр «–gpus» в «docker run».

Если вы только что установили докер, он может не работать, если вы не добавите sudo перед запуском докера.
[Ubuntu] Настройки и меры предосторожности при запуске докера без команды sudo

Это начальные настройки, необходимые для использования контейнера Docker, совместимого с графическим процессором, в Ubuntu.

Справочник