torch.
set_num_threads
Sets the number of threads used for intraop parallelism on CPU.
Warning
To ensure that the correct number of threads is used, set_num_threads must be called before running eager, JIT or autograd code.
此页内容是否对您有帮助
感谢反馈!