tf.config.PhysicalDevice
Абстракция локально видимого физического устройства.
tf.config.PhysicalDevice(
name, device_type
)
TensorFlow может использовать различные устройства, такие как процессор (CPU) или несколько графических процессоров (GPU), для вычислений. Перед инициализацией локального устройства для использования пользователь может настроить определённые свойства устройства, такие как видимость или конфигурацию памяти.
После инициализации видимого tf.config.PhysicalDevice создаётся один или несколько объектов tf.config.LogicalDevice. Используйте tf.config.set_visible_devices для настройки видимости физического устройства и tf.config.set_logical_device_configuration для настройки нескольких объектов tf.config.LogicalDevice для tf.config.PhysicalDevice. Это полезно, когда необходима изоляция между моделями или для моделирования многоустройствной среды.
Поля:
-
name: Уникальный идентификатор устройства. -
device_type: Строка, определяющая тип устройства, например, "CPU" или "GPU".
| Атрибуты | |
|---|---|
name | Псевдоним namedtuple для поля номер 0 |
device_type | Псевдоним namedtuple для поля номер 1 |
© 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/versions/r2.9/api_docs/python/tf/config/PhysicalDevice