Spec-Zone.ru › TensorFlow 1.15

tf.config.experimental.list_physical_devices

Возвращает список физических устройств, видимых для среды выполнения.

Просмотр псевдонимов

Псевдонимы для миграции

См. Руководство по миграции для получения дополнительной информации.

tf.compat.v1.config.experimental.list_physical_devices, `tf.compat.v2.config.experimental.list_physical_devices`

tf.config.experimental.list_physical_devices(
    device_type=None
)

Физические устройства — это аппаратные устройства, локально присутствующие на текущей машине. По умолчанию все обнаруженные устройства CPU и GPU считаются видимыми. list_physical_devices позволяет запросить аппаратное обеспечение до начала инициализации среды выполнения.

Следующий пример гарантирует, что машина видит как минимум 1 GPU.

physical_devices = tf.config.experimental.list_physical_devices('GPU')
assert len(physical_devices) > 0, "No GPUs found."
Аргументы
device_type (необязательно) Тип устройства для фильтрации, например, "CPU" или "GPU"
Возвращаемое значение
Список объектов PhysicalDevice

© 2020 The TensorFlow Authors. All rights reserved.
Licensed under the Creative Commons Attribution License 3.0.
Code samples licensed under the Apache 2.0 License.
https://www.tensorflow.org/versions/r1.15/api_docs/python/tf/config/experimental/list_physical_devices

Spec-Zone.ru

Настройки Оффлайн Что нового Помощь О нас
Spec-Zone .ru
спецификации, руководства, описания, API