如何优化Python代码以充分利用现代多核CPU的硬件性能?

风婷吖_8775

风婷吖_8775

2026-07-23

957人浏览

原创

python threading在cpu密集型任务中几乎无效,根本原因是cpython的gil强制同一时刻仅一个线程执行python字节码,导致多线程实为串行执行,甚至因线程切换开销而更慢;仅适用于io密集型任务,cpu密集型任务必须改用multiprocessing或processpoolexecutor。

如何优化python代码以充分利用现代多核cpu的硬件性能?

为什么 threading 在 CPU 密集型任务中几乎没用

Python 的 threading 模块跑不起来真正的并行计算,根本原因是 GIL(全局解释器锁)——同一时刻只有一个线程能执行 Python 字节码。你开 16 个 Thread 去做矩阵乘法或解析大 JSON,实际还是串行跑,甚至因线程切换更慢。

常见错误现象:top 显示 CPU 占用率卡在 100%(单核满载),其他核心闲置;任务耗时和单线程几乎一样。

  • 适用场景:仅适合 I/O 密集型任务(如 HTTP 请求、文件读写),此时线程会在等待时释放 GIL
  • 不适用场景:数值计算、图像处理、正则匹配大量文本等纯 CPU 工作
  • 替代方案:必须用 multiprocessing 或 concurrent.futures.ProcessPoolExecutor

用 ProcessPoolExecutor 替代手动管理 multiprocessing 进程

直接写 multiprocessing.Process 容易出错:进程间通信复杂、异常难捕获、资源清理漏掉导致僵尸进程。而 concurrent.futures.ProcessPoolExecutor 封装了这些细节,接口接近 ThreadPoolExecutor,迁移成本低。

实操建议:

  • 把计算函数设计成纯函数(无全局状态、无副作用),方便序列化传入子进程
  • 避免传递大型对象——executor.map() 会 pickle 参数,大数组反复序列化反而拖慢速度;优先用 numpy + shared_memory(Python 3.8+)或提前切片分发
  • 设置 max_workers:一般设为 os.cpu_count(),但若任务本身含较多 I/O 或内存带宽瓶颈,可适当调低

示例:

python-code-analyz
python-code-analyz

专业Python代码分析与优化,支持语法检查、安全扫描、性能评估、复杂度分析及重构后优化代码生成。

下载
from concurrent.futures import ProcessPoolExecutor
import os
<p>def cpu_intensive_task(x):
return sum(i * i for i in range(x))</p><p>if <strong>name</strong> == '<strong>main</strong>':
with ProcessPoolExecutor(max_workers=os.cpu_count()) as executor:
results = list(executor.map(cpu_intensive_task, [10*<em>6] </em> 4))
</p>

注意 fork vs spawn 启动方法的兼容性陷阱

Linux 默认用 fork 启动子进程,速度快但有隐患:父进程已加载的模块(尤其含 C 扩展如 torch、cv2)可能在子进程中状态异常,导致段错误或随机崩溃。macOS 和 Windows 默认用 spawn,更安全但启动慢。

解决方式:

  • 显式设置启动方法:multiprocessing.set_start_method('spawn') 放在 if __name__ == '__main__': 下方第一行
  • 必须确保所有可执行代码都包裹在 if __name__ == '__main__': 中,否则 spawn 会在子进程中重复导入执行,引发递归创建进程
  • 某些库(如旧版 PyTorch)仍对 spawn 支持不完善,需升级或查其 multiprocessing 兼容文档

别忽略数据传输开销和 NUMA 架构影响

多进程不是万能加速器。当任务本身计算量小、但参数很大(比如传一个 500MB 的 dict 给每个子进程),pickle + 反序列化时间可能远超计算时间。更隐蔽的问题是 NUMA:在多路服务器上,跨 NUMA 节点访问内存延迟高,而默认调度可能把进程分散到不同节点。

优化方向:

  • 用 multiprocessing.shared_memory 共享只读大数组(如 numpy.ndarray),避免复制
  • 用 psutil.Process().cpu_affinity() 绑定进程到特定 CPU 核心组,减少跨节点访问
  • 先做粗粒度基准测试:对比单进程跑 N 次 vs 多进程跑 N 次的总耗时,确认是否真有收益

真正卡住性能的,往往不是“没开多进程”,而是数据怎么给、给多少、给到哪。

Python免费学习笔记(深入):立即使用
在学习笔记中,你将探索 Python 的核心概念和高级技巧!

相关文章

数码产品性能查询
数码产品性能查询

该软件包括了市面上所有手机CPU,手机跑分情况,电脑CPU,电脑产品信息等等,方便需要大家查阅数码产品最新情况,了解产品特性,能够进行对比选择最具性价比的商品。

下载

本站声明:本文内容由网友自发贡献,版权归原作者所有,本站不承担相应法律责任。如您发现有涉嫌抄袭侵权的内容,请联系admin@php.cn

相关专题

更多
python打包成可执行文件
python打包成可执行文件

本专题为大家带来python打包成可执行文件相关的文章,大家可以免费的下载体验。

2023.07.20

1631

4

python能做什么
python能做什么

python能做的有:可用于开发基于控制台的应用程序、多媒体部分开发、用于开发基于Web的应用程序、使用python处理数据、系统编程等等。本专题为大家提供python相关的各种文章、以及下载和课程。

2023.07.25

3964

7

format在python中的用法
format在python中的用法

Python中的format是一种字符串格式化方法,用于将变量或值插入到字符串中的占位符位置。通过format方法,我们可以动态地构建字符串,使其包含不同值。php中文网给大家带来了相关的教程以及文章,欢迎大家前来阅读学习。

2023.07.31

1629

3

python教程
python教程

Python已成为一门网红语言,即使是在非编程开发者当中,也掀起了一股学习的热潮。本专题为大家带来python教程的相关文章,大家可以免费体验学习。

2023.08.03

22837

23

python环境变量的配置
python环境变量的配置

Python是一种流行的编程语言,被广泛用于软件开发、数据分析和科学计算等领域。在安装Python之后,我们需要配置环境变量,以便在任何位置都能够访问Python的可执行文件。php中文网给大家带来了相关的教程以及文章,欢迎大家前来学习阅读。

2023.08.04

2807

5

python eval
python eval

eval函数是Python中一个非常强大的函数,它可以将字符串作为Python代码进行执行,实现动态编程的效果。然而,由于其潜在的安全风险和性能问题,需要谨慎使用。php中文网给大家带来了相关的教程以及文章,欢迎大家前来学习阅读。

2023.08.04

2847

5

scratch和python区别
scratch和python区别

scratch和python的区别:1、scratch是一种专为初学者设计的图形化编程语言,python是一种文本编程语言;2、scratch使用的是基于积木的编程语法,python采用更加传统的文本编程语法等等。本专题为大家提供scratch和python相关的文章、下载、课程内容,供大家免费下载体验。

2023.08.11

1123

5

python合并两个列表
python合并两个列表

Python是一种强大的编程语言,具有许多方便的功能和工具。在Python中,有多种方法可以合并两个列表。php中文网给大家带来了相关的教程以及文章,欢迎大家前来学习阅读。

2023.08.10

596

4

python是前端还是后端
python是前端还是后端

Python属于前端也属于后端,其灵活性和丰富的生态系统使得开发人员能够在不同的领域中灵活运用。本专题为大家提供python相关的文章、下载、课程内容,供大家免费下载体验。

2023.08.11

2203

5

热门下载

更多
网站特效
/
网站源码
/
网站素材
/
前端模板

精品课程

更多
相关推荐
/
热门推荐
/
最新课程