在Python中检测*可用* CPU数量的便携方式

Posted 2022-09-21 pythonzhichan

根据这个问题和答案 - Python multiprocessing.cpu_count（）在4核Nvidia Jetson TK1上返回‘1‘ - Python multiprocessing.cpu_count()在某些系统上的功能输出反映了主动使用的CPU数量，而不是CPU的数量实际上可以通过调用Python程序使用。

一个常见的Python习惯用法是使用return-value cpu_count()来初始化a中的进程数Pool。但是，在使用这种“动态CPU激活”策略的系统上，该习惯用法相当严重（至少在相对静止的系统上）。

是否有一些直接（和可移植）方式来获取Python 可用处理器的数量（与当前使用的数量相反）？

笔记：

1. 这个问题是不被接受的答案来回答如何找出使用python CPU的数量，因为在上方挂了一个问题，说明这个问题，打印的内容/proc/self/status显示所有4个内核为可用的程序。

2. 在我看来，“可移植”排除了涉及解析内容的任何方法/proc/self/status，其格式可能因Linux的发行版本而异，并且在OS X上甚至不存在。（对于任何其他伪文件也是如此，同样。）

 

解决方案

---

我不认为你会得到任何真正便携的答案，所以我会给出一个正确的答案。

正确^*为Linux的回答是len(os.sched_getaffinity(pid))，哪里pid可能是0当前进程。这个函数在Python 3.3及更高版本中公开; 如果你以前需要它，你将不得不做一些奇特的cffi编码。

编辑：您可能会尝试查看是否可以使用函数（int omp_get_num_procs();如果存在），这是我在此问题上找到的唯一有意义的答案，但我还没有从Python中尝试过。

本文首发于Python黑洞网，博客园同步跟新