tf.config.experimental.set_device_policy

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

View aliases

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

Дополнительные сведения см. в руководстве по миграции.

tf.compat.v1.config.experimental.set_device_policy, `tf.compat.v2.config.experimental.set_device_policy`

tf.config.experimental.set_device_policy(
    device_policy
)

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

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

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

Args

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

device_policy

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

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

Raises
`ValueError`	Если передано недопустимое `device_policy`.

© 2020 The TensorFlow Authors. All rights reserved.
Licensed under the Creative Commons Attribution License 3.0.
Code samples licensed under the Apache 2.0 License.
https://www.tensorflow.org/versions/r1.15/api_docs/python/tf/config/experimental/set_device_policy