python - 多重处理:池和 pickle 错误 - Pickling 错误:无法 pickle :属性查找 __builtin__.instancemethod 失败

Question

我有两个文件:

x.py

class BF(object)
   def __init__():
   .
   .
   def add(self,z):
   .
   .

y.py

from y import BF
def FUNC((a,b,bf))
   .
   .
   bf.add(x)
   .
   .
   return bf

.
.
if __name__ == '__main__':
    pool = multiprocessing.Pool(3)
    for i in range(len(sl)):
         bf_set.append(BF())
    results = pool.map(FUNC,zip(sl, itertools.repeat(aa), bf_set))

我也尝试在 FUNC 中定义 BF，但我仍然得到:

PicklingError: Can't pickle <type 'instancemethod'>: attribute lookup __builtin__.instancemethod failed

我读过一些相关问题的帖子，但他们在类中有自己的 pool.map() ，所以这些解决方案不能应用于这个问题(我猜)。

有什么想法吗？

Answer 1

我将基本上使用上面的内容，但将其转换为工作代码。如果使用dill，序列化没有问题。我正在使用 multiprocessing 的 fork 叫pathos.multiprocessing ，它使用 dill而不是pickle .

>>> def FUNC((a,b,bf)):
...   z = a+b
...   bf.add(z)
...   return bf
... 
>>> class BF(object):
...   def add(self, z):
...     self.z += z
...   def __init__(self):
...     self.z = 0
... 
>>> from pathos.multiprocessing import ProcessingPool as Pool
>>> pool = Pool()
>>> 
>>> f = BF()
>>> f.add(1)
>>> f.z
1
>>> 
>>> FUNC((0,1,f))
<__main__.BF object at 0x10d387f50>
>>> 
>>> FUNC((0,1,f)).z
2
>>> 
>>> sl = [BF() for i in range(10)]
>>> results = pool.map(FUNC, zip(range(len(sl)), range(len(sl)), sl))
>>> [bf.z for bf in results]
[0, 2, 4, 6, 8, 10, 12, 14, 16, 18]

这有效，因为 pathos使用dill ，它几乎可以序列化 python 中的任何内容。

>>> import dill as pickle
>>> pickle.loads(pickle.dumps(bf.add))
<bound method BF.add of <__main__.BF object at 0x10d383950>>
>>> pickle.loads(pickle.dumps(BF.add))
<unbound method BF.add>

获取pathos和dill在:https://github.com/uqfoundation