torch.set_num_interop_threads

torch.set_num_interop_threads(int)

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

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

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