tf.config.experimental.set_visible_devices

Установите список видимых устройств.

tf.config.experimental.set_visible_devices(
    devices, device_type=None
)

Устанавливает список PhysicalDevice, которые будут помечены как видимые для выполнения. Любые устройства, которые не помечены как видимые, означают, что TensorFlow не будет выделять память на них и не сможет размещать на них операции, так как на них не будет создано ни одного LogicalDevice. По умолчанию все обнаруженные устройства помечены как видимые.

Следующий пример демонстрирует отключение первой графической карты на машине.

physical_devices = config.experimental.list_physical_devices('GPU')
assert len(physical_devices) > 0, "Not enough GPU hardware devices available"
# Disable first GPU
tf.config.experimental.set_visible_devices(physical_devices[1:], 'GPU')
logical_devices = config.experimental.list_logical_devices('GPU')
# Logical device was not created for first GPU
assert len(logical_devices) == len(physical_devices) - 1