Есть ли способ поддержки GPU без nvidia-docker

Question

Я пытаюсь получить поддержку gpu в моем контейнере без nvidia-docker

Я знаю с помощью докера nvidia, мне просто нужно использовать --runtime = nvidia, но мои текущие обстоятельства не позволяют использовать nvidia-docker

Я попытался установить драйвер nvidia, cuda, cudnn на мой контейнер, но он не работает.

Как я могу использовать gps-tenorflow без док-станции nvidia на моем контейнере?

Answer 1

Вы не можете просто установить драйверы nvidia в док-контейнер. Контейнер должен иметь доступ к оборудованию. Хотя я не уверен, но крепления могут помочь вам в этом вопросе. Смотри- https://docs.docker.com/storage/

Вы можете использовать anaconda для установки и использования Tensorflow-GPU.

Убедитесь, что у вас установлены последние версии драйверов nvidia. Установите Anaconda 2 или 3 с официального сайта. https://www.anaconda.com/distribution/

Создайте новую среду и установите tenorflow-gpu и cudatoolkit.

$conda create -n tf-gpu tensorflow-gpu python cudnn cudatoolkit

Вы также можете указать версию приложения.

Например: 1018 *

Пожалуйста, проверьте, имеет ли ваше оборудование минимальные вычислительные возможности для поддержки версии CUDA, которую вы / будете использовать.

Answer 2

Если вы не можете передать --runtime=nvidia в качестве параметра командной строки (например, docker-compose), вы можете установить время выполнения по умолчанию в файле конфигурации демона Docker /etc/docker/daemon.json:

{
    "default-runtime": "nvidia"
}

Answer 3

Вы можете использовать x11docker

Запуск образа докера на X с помощью gpu так же прост, как

x11docker --gpu imagename