Python | os.sched_setaffinity() 方法介绍

os.sched_setaffinity() 方法用于设置进程的 CPU 亲和力。这个方法仅适用于 Linux 平台。

语法

以下是os.sched_setaffinity()方法的语法:

os.sched_setaffinity(pid, mask)

参数

os.sched_setaffinity()方法接受两个参数:

• pid: 这是要设置 CPU 亲和力的进程 ID。
• mask: 这个参数接受一个 CPU 掩码序列，表示进程可以运行的 CPU。

返回值

如果成功，os.sched_setaffinity()方法将返回None。如果失败，将引发OSError异常。

实例

下面的程序演示了如何使用os.sched_setaffinity()方法，在 Linux 上设置进程的 CPU 亲和力。

# 导入 os 模块
import os

# 获取本进程的 ID
pid = os.getpid()

# 定义 CPU 掩码
mask = bytes.fromhex('01')

# 设置 CPU 亲和力
os.sched_setaffinity(pid, mask)

# 输出亲和力设置是否成功
print("CPU 亲和力设置成功")

在上面的程序中，我们首先导入了os模块，用于设置进程的 CPU 亲和力。然后我们使用os.getpid()方法来获取当前进程的 ID。接下来，我们定义一个 CPU 掩码，这个掩码在本例中将进程绑定到 CPU 核心 0 上。最后，我们调用os.sched_setaffinity()方法来设置进程的 CPU 亲和力。如果设置成功，我们将输出一条消息，表明设置成功。

注意，这个方法仅适用于 Linux 平台，并且需要 root 权限才能使用。在其他平台上，此方法将引发AttributeError异常。

以上就是Python中os.sched_setaffinity()方法的介绍。如果你需要限定你的进程使用固定的CPU核心，可以尝试使用此方法。