torch.set_num_threads

torch.set_num_threads(int)

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

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

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