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
HowTo
- Number of available GPUs:
torch.cuda.device_count()
pytorch.txt · Last modified: 2020/11/24 13:50 by pmay
