window/linux下指定核心运行python

最新推荐文章于 2023-11-03 14:37:45 发布

sa726663676

最新推荐文章于 2023-11-03 14:37:45 发布

阅读量1k

点赞数 1

分类专栏： python 文章标签：大数据

本文链接：https://blog.csdn.net/sa726663676/article/details/125545183

版权

python 专栏收录该内容

36 篇文章 0 订阅

订阅专栏

本文介绍了如何使用Python的psutil库来获取逻辑CPU数量，设置进程的CPU亲和性，从而让程序在指定的CPU核心上运行。通过实例演示了如何将当前进程绑定到特定CPU，便于优化多核系统上的任务分配。

摘要由CSDN通过智能技术生成

参考自：Python实现让程序运行在指定CPU上

# coding: utf-8
import psutil
# 获取逻辑cpu的数量
count = psutil.cpu_count()
print(f"逻辑cpu的数量是{
count}")
# Process实例化时不指定pid参数，默认使用当前进程PID，即os.getpid()
p = psutil.Process()
cpu_lst = p.cpu_affinity()
print("cpu列表", cpu_lst)
# 将当前进程绑定到cpu15上运行，列表也可以写多个cpu
p.cpu_affinity([15])  
# 运行函数消耗cpu资源

# 只需在main（）函数运行前复制前面代码即可
if __name__=="__main__":
    # 编写自己的逻辑代码即可
    pass