是否有一個pytorch方法來檢查cpus的數量？
[英]Is there a pytorch method to check the number of cpus?
問題描述
我可以使用這個torch.cuda.device_count()來檢查GPU的數量。 我想知道是否有相當於檢查CPU數量的東西。
2 個解決方案
解決方案1
2 2019-03-29 08:26:38
目前pytorch不支持DistributedDataParallel實現中的多個cpu集群。 所以，我假設您的CPU核心數量。
沒有直接等效的gpu計數方法，但你可以通過使用獲得pytorch中可用於計算的線程數
torch.get_num_threads（）
解決方案2
0 已采納 2019-03-30 15:16:51
只是用這個：
os.cpu_count()
在多個分布式 CPU 上運行 PyTorch 數據加載器/數據集
[英]Running a PyTorch dataloader/Dataset on multiple distributed CPUs
聲明:本站的技術帖子網頁，遵循CC BY-SA 4.0協議，如果您需要轉載，請注明本站網址或者原文地址。任何問題請咨詢:yoyou2525@163.com.