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/api_docs/python/tf/config/experimental/set_device_policy