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

HowTo

  • Number of available GPUs: torch.cuda.device_count()
pytorch.txt · Last modified: 2020/11/24 13:50 by pmay