torch / torch

torch.set_num_threads¶

torch.set_num_threads(int)¶: 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.

此页内容是否对您有帮助