User Tools

Site Tools


pytorch

PyTorch

How-to

Use multiple CPU for training

To use multiple CPUs for training on CPU (not GPU) you can specify torch.set_num_threads(<num_cpu>). Also see: https://pytorch.org/docs/master/generated/torch.set_num_threads.html

pytorch.txt · Last modified: 2020/07/29 11:25 by pmay