欢迎您访问程序员文章站本站旨在为大家提供分享程序员计算机编程知识!
您现在的位置是: 首页  >  IT编程

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

程序员文章站 2024-01-20 21:50:22
根据这个问题和答案 - Python multiprocessing.cpu_count()在4核Nvidia Jetson TK1上返回'1' - Python multiprocessing.cpu_count()在某些系统上的功能输出反映了主动使用的CPU数量,而不是CPU的数量实际上可以通过 ......

根据这个问题和答案 - 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黑洞网,博客园同步跟新