tf.config.set_soft_device_placement

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

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

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

Подробнее см. в руководстве по миграции.

tf.compat.v1.config.set_soft_device_placement

tf.config.set_soft_device_placement(
    enabled
)

Используется в блокнотах

Используется в руководстве
Использование GPU

Если этот параметр включен, операции могут размещаться на устройствах, отличных от явно указанных пользователем. Это может значительно снизить производительность из-за увеличения объема обмена данными между устройствами.

В некоторых случаях soft_device_placement изменяет назначение устройств:

для операции нет реализации на GPU/TPU
не обнаружено или не зарегистрировано ни одного устройства GPU
требуется совместное размещение с входными данными reftype, полученными от CPU
операцию невозможно скомпилировать с помощью XLA. Это часто встречается на TPU, для которых всегда требуется компилятор XLA.

Для TPU, если этот параметр имеет значение true, включается функция, называемая автоматической внешней компиляцией. Автоматическая внешняя компиляция переносит операции внутри программы TPU, которые невозможно скомпилировать, для выполнения на хосте. Это можно использовать при возникновении ошибок компиляции из-за неподдерживаемых операций.

Примечание: по умолчанию мягкое размещение устройств включено в режиме 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/api_docs/python/tf/config/set_soft_device_placement