Wie man die Anzahl der CPUs mit Python herausfindet

Question

Ich möchte die Anzahl der CPUs auf dem lokalen Rechner unter Verwendung von Python herausfinden. Das Ergebnis sollte user/real sein, wie es von time(1) ausgegeben wird, wenn es mit einem optimal skalierenden Benutzerbereichs-Programm aufgerufen wird.

Answer 1

Wenn Sie die Anzahl der physischen Kerne (nicht virtuelle hyperthreaded Kerne) wissen möchten, hier ist eine plattformunabhängige Lösung:

psutil.cpu_count(logical=False)

https://github.com/giampaolo/psutil/blob/master/INSTALL.rst

Beachten Sie, dass der Standardwert für logical True ist, also wenn Sie hyperthreaded Kerne einschließen möchten, können Sie folgendes verwenden:

psutil.cpu_count()

Dies wird die gleiche Anzahl liefern wie os.cpu_count() und multiprocessing.cpu_count(), bei denen keiner über das Schlüsselwortargument logical verfügt.

Answer 2

Diese geben Ihnen die Anzahl der CPU-Kerne mit Hyperthreading

1. multiprocessing.cpu_count()
2. os.cpu_count()

Diese geben Ihnen die Anzahl der virtuellen Maschinen-CPU-Kerne

1. psutil.cpu_count()
2. numexpr.detect_number_of_cores()

Nur relevant, wenn Sie mit VMs arbeiten.

Answer 3

Für Python-Versionen über 3.4 können Sie verwenden

import os
os.cpu_count()

Wenn Sie nach einem Äquivalent des Linux-Befehls nproc suchen. Sie haben diese Option

len(os.sched_getaffinity(0))

Answer 4

multiprocessing.cpu_count() gibt die Anzahl der logischen CPUs zurück, also wenn Sie eine Quad-Core-CPU mit Hyper-Threading haben, wird es 8 zurückgeben. Wenn Sie die Anzahl der physischen CPUs möchten, verwenden Sie die Python-Bindungen zu hwloc:

#!/usr/bin/env python
import hwloc
topology = hwloc.Topology()
topology.load()
print topology.get_nbobjs_by_type(hwloc.OBJ_CORE)

hwloc ist darauf ausgelegt, über verschiedene Betriebssysteme und Architekturen portabel zu sein.

Answer 5

Dies könnte für diejenigen von uns funktionieren, die verschiedene Betriebssysteme verwenden, aber das Beste aus allen Welten herausholen möchten:

import os
workers = os.cpu_count()
if 'sched_getaffinity' in dir(os):
    workers = len(os.sched_getaffinity(0))