时间戳

首页 > 解决方案 > 在 Python 中访问共享内存的段错误

python - 在 Python 中访问共享内存的段错误

问题描述

我在尝试访问共享内存时遇到了段错误，我将其归结为以下示例。f1并且f2是相同的，除了我将共享内存对象存储在一个额外的变量中f2，但是f2可以工作和f1段错误。为什么会发生这种情况？

（以相反的顺序调用这两个函数不会改变行为。）

from multiprocessing import shared_memory
import numpy as np

segment_name = "segment"
data_shape = (5,5,5)


def get_view(shm):
    shape = (1,) + data_shape
    arr_one = np.ndarray(shape, dtype=np.float64, buffer=shm.buf)
    arr_two = np.ndarray(data_shape, dtype=np.float64, buffer=arr_one[0])
    return arr_two


def f1():
    arr = get_view(shared_memory.SharedMemory(name=segment_name))
    print(arr[0][0][0])


def f2():
    shm = shared_memory.SharedMemory(name=segment_name)
    arr = get_view(shm)
    print(arr[0][0][0])


if __name__ == '__main__':
    shape = (1,) + data_shape
    dummy = np.ndarray(shape, dtype=np.float64)
    shared_memory.SharedMemory(name=segment_name, create=True, size=dummy.nbytes)
    f2()
    f1()

标签： pythonshared-memory

解决方案

事实证明，额外的参考是必要的。在不arr指向共享内存对象的情况下，del将立即调用其方法并取消链接其引用由 numpy 数组保存的底层缓冲区。

推荐阅读