是否有pytorch方法来检查CPU的数量？

Question

我可以用它torch.cuda.device_count()来检查 GPU 的数量。我想知道是否有等效于检查 CPU 数量的东西。

Answer 1

只需使用这个：

os.cpu_count()

Answer 2

目前 pytorch 在 DistributedDataParallel 实现中不支持多 cpu 集群。所以，我假设你的意思是 CPU 核心数。

gpu count 方法没有直接等效的方法，但是您可以通过使用获得可用于在 pytorch 中计算的线程数

torch.get_num_threads()