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

Question

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

Answer 1

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

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

torch.get_num_threads()

有点晚了，但值得注意的是，“get_num_threads”并没有真正获取用户可用的线程数，而是用户设置的线程数。它被初始化为正确的数字，但是如果您运行 `torch.set_num_threads(1024); print(torch.get_num_threads())`，它将显示“1024”，无论您的处理器可以支持多少个线程...更好的方法是“os.cpu_count()” (3认同)

Answer 2

只需使用这个：

os.cpu_count()