docker сборка с nvidia runtime - PullRequest
Новая
       14

docker сборка с nvidia runtime

4 голосов
/ 11 января 2020

У меня есть приложение для графического процессора, которое выполняет модульное тестирование на этапе создания образа. С Docker 19.03 можно указать время выполнения nvidia с docker run --gpus all, но мне также нужен доступ к gpus для docker build, потому что я выполняю юнит-тестирование. Как мне достичь этой цели?

Для более старой версии docker, использующей nvidia-docker2, было невозможно указать время выполнения на этапе сборки, НО вы можете установить время выполнения по умолчанию равным nvidia, и docker сборка работает нормально таким образом. Могу ли я сделать это в Docker 19.03, который больше не нуждается в nvidia- docker? Если да, то как?

Ответы [ 2 ]

4 голосов
/ 01 февраля 2020

«Решение», которое я нашел, состоит в том, чтобы сначала запустить базовый образ с установленными на нем драйверами nvidia хоста 

docker run -it --rm --gpus ubuntu

А затем вручную построить мое приложение в контейнере и зафиксировать полученное изображение. Это не идеально, и было бы лучше иметь доступ к nvidia-smi на этапе сборки.

0 голосов
/ 31 января 2020

Вы должны установить драйвер NVIDIA и Docker 19.03 для своего дистрибутива Linux. Обратите внимание, что вам не нужно устанавливать инструментарий CUDA на хост, но драйвер должен быть установлен

С В выпуске Docker 19.03 использование пакетов nvidia-docker2 не рекомендуется, поскольку графические процессоры NVIDIA теперь изначально поддерживаются в качестве устройств в среде выполнения Docker.

Если вы используете пакеты nvidia-docker2, просмотрите инструкции в « Обновление с помощью nvidia-docker2 ».

Для начинающих пользователей Docker 19.03 и графических процессоров, продолжайте с инструкциями для начала работы ниже. Использование 

# Add the package repositories
distribution=$(. /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

для получения дополнительной информации, пожалуйста, посетите: https://github.com/NVIDIA/nvidia-docker

...