Функция PyTorch torch.set_num_interop_threads
Справочное руководство PyTorch torch
torch.set_num_interop_threadsЭта функция в PyTorch используется для установки количества потоков межоперационного параллелизма. Эти потоки используются для параллельного выполнения различных независимых операций.
Определение функции
torch.set_num_interop_threads(num_threads)
Описание параметра
num_threads: Количество межоперационных потоков для установки
Пример использования
Пример
import torch
# Установить количество межоперационных потоков
torch.set_num_interop_threads(4)
print("Число межоперационных потоков установлено на:", torch.get_num_interop_threads())
# Восстановить значение по умолчанию
torch.set_num_interop_threads(1)
print("Число межоперационных потоков по умолчанию восстановлено")
# Установить количество межоперационных потоков
torch.set_num_interop_threads(4)
print("Число межоперационных потоков установлено на:", torch.get_num_interop_threads())
# Восстановить значение по умолчанию
torch.set_num_interop_threads(1)
print("Число межоперационных потоков по умолчанию восстановлено")
Результат вывода:
跨域线程数已设置为: 4 已恢复默认跨域线程数
Другие расширения