tf.config.experimental.set_device_policy

Устанавливает текущую политику устройства для потока.

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

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

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

tf.compat.v1.config.experimental.set_device_policy

tf.config.experimental.set_device_policy(
    device_policy
)

Политика устройства управляет тем, как операции, требующие ввода на определённом устройстве (например, на GPU:0), обрабатывают ввод с другого устройства (например, GPU:1).

При использовании значения по умолчанию подходящая политика выбирается автоматически. Политика по умолчанию может меняться со временем.

Эта функция устанавливает политику устройства только для текущего потока. Любой запущенный впоследствии поток снова будет использовать политику по умолчанию.

Аргументы

Аргументы
`device_policy`	Политика устройства. Допустимые значения: None: Переключиться на системную политику по умолчанию. 'warn': Копирует тензоры, которые не находятся на нужном устройстве, и выводит предупреждение. 'explicit': Вызывает ошибку, если размещение не соответствует требованиям. 'silent': Тихо копирует тензоры. Обратите внимание, что это может скрыть проблемы производительности, поскольку не предоставляется уведомление, когда операции блокируются из-за копирования тензора между устройствами. 'silent_for_int32': тихо копирует `int32` тензоры, вызывая ошибки для других.

device_policy

Политика устройства. Допустимые значения:

None: Переключиться на системную политику по умолчанию.
'warn': Копирует тензоры, которые не находятся на нужном устройстве, и выводит предупреждение.
'explicit': Вызывает ошибку, если размещение не соответствует требованиям.
'silent': Тихо копирует тензоры. Обратите внимание, что это может скрыть проблемы производительности, поскольку не предоставляется уведомление, когда операции блокируются из-за копирования тензора между устройствами.
'silent_for_int32': тихо копирует int32 тензоры, вызывая ошибки для других.

Исключения
`ValueError`	Если передан недопустимый `device_policy`.

© 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/experimental/set_device_policy