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