torch.
set_num_interop_threads
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).
此页内容是否对您有帮助
感谢反馈!