torch.set_num_threads

torch.set_num_threads(int)

Задает количество потоков, используемых для параллельных операций на CPU.

Предупреждение

Чтобы обеспечить использование правильного количества потоков, вызовите set_num_threads до выполнения кода eager, JIT или autograd.