torch / torch

torch.set_num_interop_threads¶

torch.set_num_interop_threads(int)¶: Sets the number of threads used for interop parallelism (e.g. in JIT interpreter) on CPU.

Warning

Can only be called once and before any inter-op parallel work is started (e.g. JIT execution).

此页内容是否对您有帮助