Python multiprocess pool模塊報錯pickling error問題解決方法分析

更新時間：2019年03月20日 08:59:47 作者：Arkenstone

這篇文章主要介紹了Python multiprocess pool模塊報錯pickling error問題解決方法,結(jié)合實例形式分析了multiprocess pool模塊報錯pickling error的原因與解決方法,需要的朋友可以參考下

本文實例講述了Python multiprocess pool模塊報錯pickling error問題解決方法。分享給大家供大家參考，具體如下：

問題

之前在調(diào)用class內(nèi)的函數(shù)用multiprocessing模塊的pool函數(shù)進行多線程處理的時候報了以下下錯誤信息:

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

查了下官方文檔發(fā)現(xiàn)python默認只能pickle以下的類型:

None, True, and False
integers, floating point numbers, complex numbers
strings, bytes, bytearrays
tuples, lists, sets, and dictionaries containing only picklable objects
functions defined at the top level of a module (using def, not lambda)
built-in functions defined at the top level of a module
classes that are defined at the top level of a module
instances of such classes whose dict or the result of calling getstate() is picklable (see section -
Pickling Class Instances for details).

函數(shù)只能pickle在頂層定義的函數(shù)，很明顯的class內(nèi)的函數(shù)無法被pickle因此會報錯。

import multiprocessing
def work():  # top-level 函數(shù)
  print "work!"
class Foo():
  def work(self): # 非top-level函數(shù)
    print "work"
pool1 = multiprocessing.Pool(processes=4)
foo = Foo()
pool1.apply_async(foo.work)
pool1.close()
pool1.join()
# 此時報錯
pool2 = multiprocessing.Pool(processes=4)
pool2.apply_async(work)
pool2.close()
pool2.join()
# 此時工作正常

解決方案

調(diào)用pathos包下的multiprocessing模塊代替原生的multiprocessing。pathos中multiprocessing是用dill包改寫過的，dill包可以將幾乎所有python的類型都serialize，因此都可以被pickle。或者也可以自己用dill寫一個（有點重復造輪子之嫌?。?/p>

參考

1. https://stackoverflow.com/questions/8804830/python-multiprocessing-picklingerror-cant-pickle-type-function
2. https://docs.python.org/3/library/pickle.html#what-can-be-pickled-and-unpickled
3. https://github.com/uqfoundation/pathos