博客

Python的os.sched_setaffinity()函数用于将特定进程绑定到特定的CPU核心/线程。这个函数调用需要使用Python的os库，必须运行在支持CPU集合调度的操作系统上，如Linux。

下面是使用方法的完整攻略：

1、导入os模块

首先需要在python文件中导入os模块，具体方法如下：

import os

2、获取CPU核心总数

使用下列代码获取CPU核心总数，以判断CPU复杂度：

cores = os.cpu_count()

3、创建CPU核心集合

创建一组位于0和cores（即CPU总核心数）之间的核心索引的列表，例如将要绑定进程到0和1号核心，则集合为：

cpu_affinity = [0, 1]

4、绑定进程到特定的CPU核心上

使用如下方式将指定进程ID与特定的CPU核心集合相关联：

process_id = 12345   # 进程ID
os.sched_setaffinity(process_id, cpu_affinity)

5、完整代码

整个代码如下所示：

import os

cores = os.cpu_count()
cpu_affinity = [0, 1]
process_id = 12345   # 进程ID
os.sched_setaffinity(process_id, cpu_affinity)

需要说明的是，os.sched_setaffinity()函数需要运行在支持CPU核心集合调度的操作系统上，例如Linux。如果运行在不支持CPU集合调度的操作系统上，则会引发AttributeError异常。

另外，绑定进程到特定的CPU核心上可能会对系统性能产生影响，必须慎重考虑和测试。