首页 > 解决方案 > 有没有pytorch方法来检查cpu的数量?

问题描述

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

标签: pythonneural-networkdeep-learningpytorch

解决方案


就用这个:

os.cpu_count()

推荐阅读