Блог

Главная - Вопросы по программированию - Использует ли изображение tensorflow в docker графический процессор?

Использует ли изображение tensorflow в docker графический процессор?

#python #docker #tensorflow

#python #docker #tensorflow

Вопрос:

У меня вопрос, использует ли мой docker графический процессор.

На моем рабочем столе установлены GeForce 2060 super и Ryzen 5 3600.

Я реализовал CNN с помощью tensorflow. Каждое изображение имеет от 20 до 50 КБ.

Когда я использую только процессор, каждая эпоха тратит 23 секунды (около 95 мс / шаг). С другой стороны, когда я использую изображение tensorflor-gpu-jupyter в docker, каждая эпоха тратит 21 секунду (около 83 мс / шаг). Моя команда docker выглядит следующим образом.

 docker run --gpus all --rm -p 8888:8888 -v $PWD/CNN:/tf/notebooks tensorflow/tensorflow:latest-gpu-jupyter

Я слышал, что графический процессор ускоряет его в 5-10 раз быстрее,
но для меня это не так.

Использует ли мой docker графический процессор?

К вашему сведению. Я проверил, что nvidia работает со следующей командой.

 docker run --gpus all --rm nvidia/cuda nvidia-smi

Ответ №1:

В официальных изображениях TensorFlow для Docker включен графический процессор, если хост-система правильно настроена (исходный код). Вы должны извлекать изображения с -gpu помощью тега.

Самый простой способ проверить: использовать nvtop или nvidia-smi -l 10 проверить использование графического процессора в хост-системе.

Обратите внимание, что для включения поддержки графического процессора у вас должен быть установлен NVIDIA Container Toolkit.

Комментарии:

1. С помощью команды nvidia-smi -l 10 он показал «Использование памяти 7884 МБ / 7981, загрузка графического процессора 4%» во время запуска python. Работает ли графический процессор?

2. Хм… Я не мог сказать. Выделенный так много памяти определенно является признаком того, что что-то происходит, но использование графического процессора низкое. Может быть связано с использованием инфраструктуры или с тем, что вы выполнили проверку в цикле ожидания. Я настоятельно рекомендую установить nvtop и проверить там, поскольку он покажет процессы, использующие графический процессор, чтобы вы точно знали, работает он или нет.

3. Я пытаюсь запустить nvtop, и он показал Impossible to initialize nvidia nvml : Driver/library version mismatch . Я думаю, что это приводит к тому, что графический процессор не работает должным образом. Я переустановлю nvidia и попробую это снова.