### Python 中设置和获取 CPU 亲和性的方法 在 Python 中，可以通过 `os` 模块中的 `sched_setaffinity` 和 `sched_getaffinity` 函数来设置或获取进程的 CPU 亲和性。这些函数允许指定哪些 CPU 核心可以运行特定的进程或线程。 以下是具体的方法说明： #### 获取当前进程的 CPU 亲和性 使用 `os.sched_getaffinity(pid)` 方法可以获得给定进程 ID 的 CPU 亲和性集合。如果参数 pid 是 0，则表示操作的是当前进程[^3]。 ```python import os # 获取当前进程的 CPU 亲和性 current_affinity = os.sched_getaffinity(0) print(f"Current CPU Affinity Mask: {current_affinity}") ``` #### 设置当前进程的 CPU 亲和性 使用 `os.sched_setaffinity(pid, mask)` 方法可以设置给定进程 ID 的 CPU 亲和性集合。同样，当 pid 为 0 时，表示修改当前进程的 CPU 亲和性。 ```python import os # 定义新的 CPU 亲和性掩码 (假设只希望进程运行在核心 1 和 核心 2 上) new_mask = {1, 2} # 设置当前进程的 CPU 亲和性 os.sched_setaffinity(0, new_mask) # 验证新设置的 CPU 亲和性 updated_affinity = os.sched_getaffinity(0) print(f"Updated CPU Affinity Mask: {updated_affinity}") ``` 需要注意的是，在某些操作系统上可能需要管理员权限才能成功更改 CPU 亲和性配置。 #### 使用第三方库扩展功能 除了标准库外，还可以借助像 `psutil` 或者专门针对 CPU 亲和力设计的小型工具库（如 `affinity`）。虽然两者都可以用来管理 CPU 亲和性，但是它们之间存在一些差异[^2]: - **psutil**: 提供了跨平台支持，并且能够轻松访问系统的各种资源状态信息。 ```python import psutil p = psutil.Process() # 查看当前进程可用的核心列表 current_cpus = p.cpu_affinity() print(f"PSUtil Current CPUs: {current_cpus}") # 修改可使用的 CPU 列表 p.cpu_affinity([1, 2]) updated_cpus = p.cpu_affinity() print(f"PSUtil Updated CPUs: {updated_cpus}") ``` - **affinity**: 更专注于提供简单易用的方式来调整单个或者多个线程/进程级别的 CPU 绑定关系。 无论选择哪种方式，都应考虑目标环境的操作系统兼容性和性能需求等因素后再做决定。 ---