tf.config.set_soft_device_placement

Включить или отключить мягкое размещение устройств.

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

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

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

tf.compat.v1.config.set_soft_device_placement

tf.config.set_soft_device_placement(
    enabled
)

Если включено, операция будет размещена на ЦП, если выполняется хотя бы одно из следующих условий:

нет реализации операции на GPU
нет известных или зарегистрированных устройств GPU
необходимо разместить вместе с входными данными типа ref, которые поступают с ЦП

Примечание: по умолчанию мягкое размещение устройств включено при выполнении в режиме eager (для удобства) и отключено в режиме графа (для производительности).

Аргументы
`enabled`	Логическое значение, указывающее, нужно ли включить мягкое размещение.

© 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/set_soft_device_placement