通过研究案例,彻底掌握Python GIL

2023年 11月 4日 52.5k 0

Python因其全局解释器锁(GIL)而声名狼藉。GIL限制了Python解释器一次只能执行一个线程。在现代多核CPU上,这是一个问题,因为程序无法利用多个核心。不过,尽管存在这种限制,Python仍已成为从后端Web应用到AI/ML和科学计算等领域的顶级语言。

1、训练数据管道的结构

对于大多数后端Web应用来说,GIL的限制并不是一个约束,因为它们通常受到I/O的限制。在这些应用中,大部分时间只是等待来自用户、数据库或下游服务的输入。系统只需具备并发性,而不一定需要并行性。Python解释器在执行I/O操作时会释放GIL,因此当线程等待I/O完成时,就会给另一个线程获得GIL并执行的机会。

GIL的限制不会影响大多数计算密集型的AI/ML和科学计算工作负载,因为像NumPy、TensorFlow和PyTorch等流行框架的核心实际上是用C++实现的,并且只有Python的API接口。大部分计算可以在不获取GIL的情况下进行。这些框架使用的底层C/C++内核库(如OpenBLAS或Intel MKL)可以利用多个核心而不受GIL的限制。

当同时有I/O和计算任务时会发生什么?

2、使用纯Python的计算任务

具体来说,可以考虑以下两个简单的任务。

import time

def io_task():
    start = time.time()
    while True:
        time.sleep(1)
        wake = time.time()
        print(f"woke after: {wake - start}")
        start = wake
        
def count_py(n):
  compute_start = time.time()
  s = 0
  for i in range(n):
      s += 1
  compute_end = time.time()
  print(f"compute time: {compute_end - compute_start}")
  return s

在这里,通过休眠一秒钟来模拟一个I/O限制的任务,然后唤醒并打印它休眠了多长时间,然后再次休眠。count_py是一个计算密集型的任务,它简单地对数字n进行计数。如果同时运行这两个任务会发生什么?

import threading

io_thread = threading.Thread(target=io_task, daemnotallow=True)
io_thread.start()
count_py(100000000)

输出结果如下:

woke after: 1.0063529014587402
woke after: 1.009704828262329
woke after: 1.0069530010223389
woke after: 1.0066332817077637
compute time: 4.311860084533691

count_py需要大约4.3秒才能计数到一百万。但是io_task在同一时间内运行而不受影响,大约在1秒后醒来,与预期相符。尽管计算任务需要4.3秒,但Python解释器可以预先从运行计算任务的主线程中释放GIL,并给予io_thread获得GIL并运行的机会。

3、使用numpy的计算任务

现在,本文将在numpy中实现计数函数,并进行与之前相同的实验,但这次要计数到一千万,因为numpy的实现效率更高。

import numpy as np

def count_np(n):
    compute_start = time.time()
    s = np.ones(n).sum()
    compute_end = time.time()
    print(f"compute time: {compute_end - compute_start}")
    return s
  
io_thread = threading.Thread(target=io_task, daemnotallow=True)
io_thread.start()
count_np(1000000000)

输出结果如下:

woke after: 1.0001161098480225
woke after: 1.0008511543273926
woke after: 1.0004539489746094
woke after: 1.1320469379425049
compute time: 4.1334803104400635

这显示的结果与上一次实验类似。在这种情况下,不是Python解释器预先释放了GIL,而是numpy自己主动释放了GIL。

这是否意味着在独立的线程中同时运行I/O任务和计算任务总是安全的?

4、使用自定义C++扩展的计算任务

现在,本文将用Python的C++扩展实现计数函数。

// importing Python C API Header
#include
#include

static PyObject *count(PyObject *self, PyObject *args){
long num;

if (!PyArg_ParseTuple(args, "l", &num))
return NULL;
long result = 0L;
std::vector v(num, 1L);
for (long i=0L; i

相关文章

JavaScript2024新功能:Object.groupBy、正则表达式v标志
PHP trim 函数对多字节字符的使用和限制
新函数 json_validate() 、randomizer 类扩展…20 个PHP 8.3 新特性全面解析
使用HTMX为WordPress增效:如何在不使用复杂框架的情况下增强平台功能
为React 19做准备:WordPress 6.6用户指南
如何删除WordPress中的所有评论

发布评论