Windows上使用Python绑定CPU

最新推荐文章于 2024-07-06 02:33:12 发布

原创最新推荐文章于 2024-07-06 02:33:12 发布 · 2.9k 阅读

2 ·

CC 4.0 BY-SA版权

Python 专栏收录该内容

82 篇文章

订阅专栏

本文介绍如何使用Python在Windows平台下通过API将进程绑定到特定的CPU核心上运行，包括获取当前进程信息、查询CPU数量及具体绑定方法。

部署运行你感兴趣的模型镜像

在 Windows 平台上，有时候我们想让自己的程序在指定的CPU上运行，此时我看可以通过下面几个API来实现

win32process.SetProcessAffinityMask(hProcess, mask) - 绑定进程到CPU
win32api.GetSystemInfo()[5] - 查询当前机器有几个CPU
win32api.GetCurrentProcess() - 获取当前进程信息

其中绑定CPU是mask的值如下：

0x0001 1
0x0002 2
0x0003 1 or 2
0x0004 3
0x0005 1 or 3
0x0006 2 or 3
0x0007 1, 2, or 3
0x000F 1, 2, 3, or 4
...

示例

import win32process
import win32api

# 绑定到CPU 1
win32process.SetProcessAffinityMask(win32api.GetCurrentProcess(), 0x0001)

# 绑定到CPU 2
win32process.SetProcessAffinityMask(win32api.GetCurrentProcess(), 0x0002)

# 绑定到CPU 1 or 2
win32process.SetProcessAffinityMask(win32api.GetCurrentProcess(), 0x0003)