torch.set_num_interop_threads

torch.set_num_interop_threads(int)

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

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

Можно вызвать только один раз и до начала любых параллельных работ взаимодействия (например, выполнения JIT).