Есть ли у Docker образ, поддерживающий nvidia gpu на процессоре?

Question

Я хочу запустить глубокую нейронную сеть на контейнере Docker.Я могу использовать изображения TensorFlow, PyTorch или CNTK, доступные публично на Docker Hub, которые поддерживают графический процессор.И я могу получить графический процессор NVIDA в облаке.

Что происходит, когда на моем ПК запускаются контейнеры на основе графического процессора, в которых нет графического процессора (или графического процессора Intel)?Будет ли контейнер Docker на базе графического процессора NVIDIA корректно работать на моем процессоре (хотя и медленнее)?

Было бы здорово, если бы я мог разрабатывать локально без графического процессора и увеличивать масштаб с помощью графического процессора, когда я буду готов к работе наscale.

Будет ли это работать?Мысли?

Answer 1

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

Если вы хотите, чтобы TensorFlow автоматически выбирал существующее и поддерживаемое устройство для запуска операций, если указанное несуществует, вы можете установить для allow_soft_placement значение True в параметре конфигурации при создании сеанса.

# Creates a graph.
with tf.device('/device:GPU:2'):
  a = tf.constant([1.0, 2.0, 3.0, 4.0, 5.0, 6.0], shape=[2, 3], name='a')
  b = tf.constant([1.0, 2.0, 3.0, 4.0, 5.0, 6.0], shape=[3, 2], name='b')
  c = tf.matmul(a, b)
# Creates a session with allow_soft_placement and log_device_placement set
# to True.
sess = tf.Session(config=tf.ConfigProto(
      allow_soft_placement=True, log_device_placement=True))
# Runs the op.
print(sess.run(c))

Answer 2

Если Tensorflow не найдет графический процессор, он будет использовать только процессор, поэтому это должно работать без проблем.