torch.set_num_threads

torch.set_num_threads(int)

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

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

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