python中affinity库和psutil库区别以及设置cpu亲和力

psutil 和 affinity 都是用于处理进程相关任务的 Python 库，但它们的功能范围和主要用途有所不同。

0. psutil（Python System and Process Utilities）

主要功能：

• psutil 是一个用于系统监控、进程管理和资源使用情况的跨平台库。
• 它提供了获取系统信息的功能，包括 CPU、内存、磁盘、网络的使用情况，以及进程的信息和管 - 理。

特点：

• 跨平台：psutil 支持 Windows、Linux、macOS 和其他平台。
• 丰富功能：除了设置 CPU 亲和性，它还可以获取进程 CPU、内存、网络等的详细信息，并且可以监控系统整体资源使用情况。
• API 稳定且文档完善，应用范围广泛。

功能示例：

• 获取进程列表。
• 获取 CPU、内存、磁盘、网络的使用信息。
• 监控进程，获取进程的 CPU 使用率、内存占用等。
• 设置进程 CPU 亲和性。

  import psutil
  # 获取当前进程
  process = psutil.Process()
  # 获取当前进程的 CPU 亲和性
  cpu_affinity = process.cpu_affinity()
  # 设置进程的 CPU 亲和性到 CPU 0 和 1
  process.cpu_affinity([0, 1])
  

优点：

• 跨平台支持。
• 功能广泛，不仅局限于 CPU 亲和性。

缺点：

• 作为一个功能丰富的库，如果只需要简单的 CPU 亲和性设置，可能显得有些“重量级”。

1. affinity

主要功能：

• affinity 是一个用于设置和管理进程的 CPU 亲和性（CPU affinity）的库。
• 亲和性即限制进程只能在某些指定的 CPU 上运行，这样可以提升性能或者实现特定的进程控制。

特点：

• 专注于 CPU 亲和性：相比 psutil，affinity 主要功能就是设置进程的 CPU 亲和性。
• 支持多进程亲和性控制：你可以针对不同的进程设定不同的 CPU 亲和性。

功能示例：

• 设置一个进程只能在特定的 CPU 核心上运行。

  import affinity
  
  # 设置当前进程只在 CPU 0 和 CPU 1 上运行
  affinity.set_process_affinity_mask(0, 0b11)
  

优点：

• 简单且专注：affinity 是一个非常轻量的库，只负责 CPU 亲和性相关的功能。
• 对于需要精确控制 CPU 亲和性的程序，它非常有效。

缺点：

• 不支持跨平台：affinity 仅支持 Linux 和 Windows 系统，并且由于部分语法，它可能不完全兼容 Python 3。
• 功能较单一，仅用于 CPU 亲和性设置，没有其他的系统或进程监控功能。

总结对比

适用场景

• 如果你只需要设置进程的 CPU 亲和性，并且在 Linux 或 Windows 系统上运行，affinity 是一个轻量且专注的选择。
• 如果你需要获取系统或进程的广泛信息，并且需要一个跨平台的解决方案，psutil 是更合适的工具，尽管它可能显得“重”一些。