迹忆客 专注技术分享

CPU 可以包含单核或多核。 单核只处理一个进程，而多核同时处理多个进程。

本篇文章将介绍使用 Python 程序查找 CPU 内核总数的不同方法。

使用 multiprocessing 模块获取 Python 中的 CPU 数量

multiprocessing 模块中的 cpu_count() 函数获取系统中的 CPU 总数。

import multiprocessing
multiprocessing.cpu_count()

输出:

8

使用 os 模块获取 Python 中的 CPU 数量

os 模块提供与操作系统交互的功能。 os.cpu_count() 命令可以打印系统中的 CPU 数量。

该方法与上述示例相同，在同一台机器上返回相同的值。

import os
os.cpu_count()

输出:

8

此输出值不等于当前进程可以使用的 CPU 数。

您可以使用以下命令获取可用 CPU 的数量。

len(os.sched_getaffinity(0))

它只适用于某些操作系统，例如 Unix。

在 Python 中使用 psutil 模块获取 CPU 数量

psutil 是 Python 中的一个跨平台模块，用于获取有关正在运行的进程和系统利用率（CPU、内存、磁盘、网络、传感器）的信息。

您可以使用以下命令安装 psutil 库。

pip install psutil

要使用 psutil 确定 CPU 的数量，请执行这些命令。

import psutil
psutil.cpu_count()

输出:

8

使用 joblib 模块获取 Python 中的 CPU 数量

joblib 是一个 Python 包，它提供用于透明和快速的函数磁盘缓存和简单并行计算的工具。

您可以使用 joblib.cpu_count() 命令查看系统中的 CPU 数量。

import joblib
joblib.cpu_count()

输出:

8

现在你应该明白了如何在 Python 中获取使用不同模块的 CPU 总数。 所有模块都有相同的函数 cpu_count()，它返回系统中的 CPU 计数。

您还学习了如何找出计算机中可用 CPU 的数量。 我们希望您觉得本篇文章对您有所帮助。