tf.config.set_visible_devices

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

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

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

Следующий пример демонстрирует отключение первой GPU на компьютере.

physical_devices = tf.config.list_physical_devices('GPU')
try:
  # Disable first GPU
  tf.config.set_visible_devices(physical_devices[1:], 'GPU')
  logical_devices = tf.config.list_logical_devices('GPU')
  # Logical device was not created for first GPU
  assert len(logical_devices) == len(physical_devices) - 1
except:
  # Invalid device or cannot modify virtual devices once initialized.
  pass