Есть ли метод pytorch для проверки количества процессоров?

Question

Я могу использовать это torch.cuda.device_count() для проверки количества графических процессоров.Мне было интересно, есть ли что-то эквивалентное, чтобы проверить количество процессоров.

Answer 1

В настоящее время pytorch не поддерживает кластер с несколькими процессорами в реализации DistributedDataParallel. Итак, я предполагаю, что вы имеете в виду количество ядер процессора.

Прямого эквивалента для метода подсчета gpu нет, но вы можете получить количество потоков, доступных для вычисления в pytorch, с помощью

torch.get_num_threads ()

Answer 2

просто используйте это:

os.cpu_count()