tf.config.PhysicalDevice
Абстракция физического устройства, доступного локально.
tf.config.PhysicalDevice(
name, device_type
)
TensorFlow может использовать для вычислений различные устройства, например ЦП или несколько графических процессоров. Перед инициализацией локального устройства пользователь может настроить некоторые его свойства, например видимость или конфигурацию памяти.
После инициализации видимого устройства 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/api_docs/python/tf/config/PhysicalDevice