多处理numpy未定义错误

Question

我正在使用以下测试代码：

from pathos.multiprocessing import ProcessingPool as Pool
import numpy
def foo(obj1, obj2):
   a = obj1**2
   b = numpy.asarray(range(1,5))
   return obj1, b
if __name__ == '__main__':
    p = Pool(5)
    res = p.map(foo, [1,2,3], [4,5,6])

它给出错误：

File "C:Python27libsite-packagesmultiprocesspool.py", line 567, in get
    raise self._value
NameError: global name 'numpy' is not defined

我在代码中做错了什么？

编辑：为什么这个问题被否决了两次？

我已经安装了numpy，我的解释器一直在正确使用它，直到我尝试为多处理进行它。我已经用相同的安装进行了一段时间的编码。

Answer 1

导入似乎不是在进程之间共享的。因此，您需要在所有流程中分别使用import numpy。

在您的情况下，这意味着在foo函数中添加import numpy。进程不是轻量级的，所以import不会降低速度(至少不会显著降低速度(。

另一种选择是将模块传递给功能(不推荐，我不确定这是否有效(：

if __name__ == '__main__':
    p = Pool(5)
    res = p.map(foo, numpy, [1,2,3], [4,5,6])
def foo(np, obj1, obj2):
   a = obj1**2
   b = np.asarray(range(1,5))
   return obj1, b