如何判断所用Linux版本是否支持fork()的写时复制功能？

嘿，这个问题我太有共鸣了——当初处理大内存数据集用多进程的时候，也担心过子进程会把整个数据结构复制一遍，直接把内存榨干！好在现在主流Linux都支持COW，下面给你几个简单的验证手段，快速确认：

1. 先看内核版本（最快捷） #

Linux的写时复制机制从2.3.20版本开始就默认启用了，所以先查你的内核版本：

uname -r

如果输出的版本号大于等于2.3.20（比如常见的3.x、4.x、5.x、6.x系列），那基本可以确定系统支持COW。现在市面上几乎所有在用的Linux发行版（Ubuntu、CentOS、Debian等）的内核都远高于这个版本，除非你用的是极其老旧的系统。

2. 用Python脚本实际测试内存占用 #

最直观的方式是写个小脚本，创建大内存数据后fork子进程，观察父子进程的实际物理内存使用：

import multiprocessing
import psutil
import numpy as np

def idle_worker():
    # 子进程挂起，等待输入再退出
    input("Press Enter to terminate worker process...")

if __name__ == "__main__":
    # 创建一个2GB的大型数组（根据你的内存调整大小）
    large_dataset = np.ones((500_000_000,), dtype=np.float32)  # 500M元素 ×4字节=2GB
    main_rss = psutil.Process().memory_info().rss / 1024**2
    print(f"主进程常驻内存（RSS）: {main_rss:.2f} MB")

    # 启动子进程
    child_proc = multiprocessing.Process(target=idle_worker)
    child_proc.start()
    
    child_rss = psutil.Process(child_proc.pid).memory_info().rss / 1024**2
    print(f"子进程常驻内存（RSS）: {child_rss:.2f} MB")

    child_proc.join()

• 如果COW生效：子进程的RSS会远小于主进程的2GB（可能只有几十MB），因为子进程只是共享主进程的内存页，没有实际复制。
• 如果COW未生效：子进程的RSS会接近主进程的内存占用，说明整个数据结构被复制了。

3. 用pmap工具查看内存页状态 #

可以通过pmap工具深入查看进程的内存页细节：

1. 运行上面的脚本，等子进程启动后，记下子进程的PID。
2. 执行命令：

pmap -x <子进程PID>

查看输出中的Dirty列——COW机制下，未被修改的共享内存页的Dirty值为0；只有当子进程修改了某页数据后，该页的Dirty值才会增加，代表这部分内存已经被复制到子进程的私有空间。

补充说明 #

在Linux上，multiprocessing.Pool默认使用fork方式创建子进程，所以完全能享受到COW的好处——只有当子进程修改共享数据时，才会复制对应的内存页，大大降低了内存开销。除非你手动指定了spawn或forkserver启动方式（这两种不会共享父进程内存）。

内容的提问来源于stack exchange，提问作者Rob Streeting