torch.set_num_interop_threads

torch.set_num_interop_threads(int)

设置 CPU 上用于交互并行（例如在 JIT 解释器中）的线程数。

警告

只能调用一次，并且在任何交互并行工作开始之前（例如 JIT 执行）。