tf.config.get_visible_devices
Получает список видимых физических устройств.
tf.config.get_visible_devices(
device_type=None
)
Возвращает список объектов PhysicalDevice, помеченных как видимые для среды выполнения. После инициализации среды выполнения с каждым видимым устройством будет связан как минимум один объект LogicalDevice.
В следующем примере проверяется, что все видимые графические процессоры отключены:
physical_devices = tf.config.list_physical_devices('GPU')
try:
# Disable all GPUS
tf.config.set_visible_devices([], 'GPU')
visible_devices = tf.config.get_visible_devices()
for device in visible_devices:
assert device.device_type != 'GPU'
except:
# Invalid device or cannot modify virtual devices once initialized.
pass| Аргументы | |
|---|---|
device_type | (необязательная строка) Включает только устройства указанного типа. Например, «CPU» или «GPU». |
| Возвращает | |
|---|---|
Список видимых объектов PhysicalDevice |
© 2022 The TensorFlow Authors. All rights reserved.
Licensed under the Creative Commons Attribution License 4.0.
Code samples licensed under the Apache 2.0 License.
https://www.tensorflow.org/api_docs/python/tf/config/get_visible_devices