torch.set_num_interop_threads

torch.set_num_interop_threads(int)

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

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

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