python如何指定使用的cpu核_python设置CPU亲缘性

由于未知的原因，一个python程序总是占用一个CPU核心，导致一个使用率满的，另一个很低，不均衡。固然可以在任务管理中设置进程相关性为所有处理器，但是每次都设置太过麻烦，查了下可以使用windows的API设置，但是要在python中设置，需要做点工作。

其次，下载affinity模块安装，这里提供了访问

然后就可以使用了。举例如下，在程序的开始地方加上代码

1 import affinity #need pywin32

2 importos3

4 affinity.set_process_affinity_mask(os.getpid(), 3L) #set all cpu to run

运行程序，在任务管理器中可以看到程序在使用两个CPU。

说明：程与指定cpu绑定：SetProcessAffinityMask(GetCurrentProcess(), dwMask);

dwMask为CPU序号的或运算值：1(0001)代表只运行在CPU1，2(0010)代表只运行在CPU2，3(0011)代表可以运行在CPU1和CPU2，以此类推。

还没找到python获取CPU核心数量的函数，所以手工查看一下了，貌似win32con可以做到，linux下也没研究。。。

附：

CPU亲缘性介绍

按照默认设置，当系统将线程分配给处理器时，Windows使用软亲缘性来进行操作。这意味着如果所有其他因素相同的话，它将设法在它上次运行的那个处理器上运行线程。让线程留在单个处理器上，有助于重复使用仍然在处理器的内存高速缓存中的数据。