tf.config.LogicalDeviceConfiguration
Класс конфигурации для логических устройств.
tf.config.LogicalDeviceConfiguration(
memory_limit=None, experimental_priority=None
)
Класс определяет параметры для конфигурации tf.config.PhysicalDevice, когда он инициализируется как tf.config.LogicalDevice во время запуска инициализации. Не все поля допустимы для всех типов устройств.
См. tf.config.get_logical_device_configuration и tf.config.set_logical_device_configuration для примеров использования.
Поля:
-
memory_limit: (необязательно) Максимальный объем памяти (в МБ), который нужно выделить на виртуальном устройстве. В настоящее время поддерживается только для графических процессоров. -
experimental_priority: (необязательно) Приоритет, назначаемый виртуальному устройству. Более низкие значения имеют более высокий приоритет, а 0 — значение по умолчанию. В пределах физического графического процессора планировщик графического процессора отдаст приоритет операциям на виртуальных устройствах с более высоким приоритетом. В настоящее время поддерживается только для графических процессоров Nvidia.
| Атрибуты | |
|---|---|
memory_limit | |
experimental_priority | |
© 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/r2.3/api_docs/python/tf/config/LogicalDeviceConfiguration