Включить поддержку gpu по умолчанию для контейнеров docker - PullRequest
0 голосов
/ 12 апреля 2020

Я использую платформу ( Cytomine ) в Ubuntu 18.04 для запуска некоторых глубоко изученных контейнерных приложений (эта платформа автоматически обрабатывает Docker изображений и контейнеров, поэтому мне нужно только создать изображение и предоставить его URL для загрузки на платформу). Пока все работает хорошо, но теперь мне нужно включить поддержку GPU для эффективной работы модели. Таким образом, я провел несколько локальных тестов с nvidia- docker, чтобы вручную запустить контейнер модели с поддержкой графического процессора, было действительно легко заставить его работать, потому что мне просто нужно было добавить один параметр в команду run:

docker run --gpus all

Однако, поскольку я не могу добавить эту опцию в код на платформе Cytomine , мне нужно найти способ добавить / включить эту опцию по умолчанию для всех контейнеров, запущенных docker.

Я попытался добавить эту опцию в файлы /etc/docker/daemon.json and /etc/docker/key.json, а затем перезапустил docker sudo systemctl restart docker. Однако это не сработало.

Также я нашел способ создания docker конфигурационных файлов (docker config ); однако, похоже, что это работает только с Docker Swarm, и я не собираюсь использовать Swarm для этого проекта.

Таким образом, я ищу простое решение, которое можно правильно развернуть. Есть ли способ включить эту опцию (--gpus all) по умолчанию при запуске любого контейнера Docker? (как-нибудь включить его в Dockerfile?)

Спасибо!

Добро пожаловать на сайт PullRequest, где вы можете задавать вопросы и получать ответы от других членов сообщества.
...