torch.init_num_threads

torch.init_num_threads() → None

init_num_threads()

Инициализирует количество параллельных потоков, используемых в текущем потоке.

Вызывайте эту функцию при создании каждого нового потока, чтобы передать значения из torch.set_num_threads() в новый поток.