Функция config.set_soft_device_placement

Функция config.set_soft_device_placement управляет режимом мягкого размещения операций на устройствах. Первым параметром функция принимает булево значение: True включает мягкое размещение, а False отключает его. При включенном режиме TensorFlow автоматически выбирает доступное устройство для выполнения операции, если указанное устройство недоступно. По умолчанию мягкое размещение включено.

Синтаксис

tf.config.set_soft_device_placement(enable)

Пример

Давайте включим мягкое размещение и проверим текущее состояние через внутренний флаг:

import tensorflow as tf

tf.config.set_soft_device_placement(True)
print(tf.config.get_soft_device_placement())

Результат выполнения кода:

True

Пример

Давайте отключим мягкое размещение и посмотрим, как изменится флаг состояния:

import tensorflow as tf

tf.config.set_soft_device_placement(False)
print(tf.config.get_soft_device_placement())

Результат выполнения кода:

False

Пример

Давайте создадим тензор и выполним операцию при включенном мягком размещении:

import tensorflow as tf

tf.config.set_soft_device_placement(True)

t = tf.constant([1, 2, 3, 4, 5])
res = t * 2
print(res)

Результат выполнения кода:

tf.Tensor([ 2 4 6 8 10], shape=(5,), dtype=int32)

Смотрите также

функцию config.list_physical_devices,
которая возвращает список физических устройств
функцию config.set_visible_devices,
которая задает видимые устройства
функцию device,
которая задает контекст устройства для операций
функцию config.get_visible_devices,
которая возвращает список видимых устройств