torch.init_num_threads

torch.init_num_threads() → None

init_num_threads()

Initializes the number of parallel threads used on the current thread.

Call this whenever a new thread is created in order to propagate values from torch.set_num_threads() onto the new thread.