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