Python multiprocessing.cpu_count() 在 4 核 Nvidia Jetson TK1 上返回“1”

Question

谁能告诉我为什么 Python 的multiprocessing.cpu_count()函数在具有四个 ARMv7 处理器的 Jetson TK1 上调用时会返回1 ？

>>> import multiprocessing
>>> multiprocessing.cpu_count()
1

Jetson TK1 开发板或多或少是直接开箱即用的，没有人弄乱过 cpuset。 在同一个 Python shell 中，我可以打印/proc/self/status ，它告诉我该进程应该可以访问所有四个核心：

>>> print open('/proc/self/status').read()
----- (snip) -----
Cpus_allowed:   f
Cpus_allowed_list:      0-3
----- (snip) -----

还有什么可能导致cpu_count()出现这种行为？

编辑：

为了检验 Klaus 的假设，我使用以下代码运行了一个非常简单的实验：

import multiprocessing

def f(x):
    n = 0
    for i in xrange(10000):
        n = max(n, multiprocessing.cpu_count())
    return n

p = multiprocessing.Pool(5)
for i in range(10):
    print p.map(f, [1,2,3,4,5])

这产生了以下输出：

[3, 3, 3, 3, 1]
[4, 3, 3, 3, 3]
[4, 3, 3, 3, 3]
[3, 3, 4, 3, 3]
[4, 3, 3, 3, 3]
[3, 3, 4, 3, 3]
[4, 3, 3, 3, 3]
[3, 3, 4, 3, 3]
[3, 3, 3, 4, 3]
[4, 3, 3, 3, 3]

仅运行p.map(f, [1,2,3,4,5])的单次迭代通常会产生[1, 1, 1, 1, 1] ，尽管偶尔会出现2作为列表元素之一.

Answer 1

在 Linux 系统上， multiprocessing.cpu_count()依赖于sysconf (_SC_NPROCESSORS_ONLN)调用，它返回在线CPU 的数量，而sysconf (_SC_NPROCESSORS_CONF)则返回已配置的CPU 的数量。

在具有高级 CPU 电源管理功能（将 CPU 内核设置为离线以节省能源）或具有类似动态 CPU 激活功能的系统中，这些值可能会有所不同。

Answer 2

os.cpu_count() 的文档（声明它返回 CPU 总数，而不是可用 CPU 的数量）提供了一种计算可用 CPU 的方法：

len(os.sched_getaffinity(0))

见https://docs.python.org/3/library/os.html#os.cpu_count