pytorch
Table of Contents
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
