使用自訂函數進行多重處理時如何解決 AttributeError？

Multiprocessing Issue: Resolving the AttributeError

When attempting to implement multiprocessing in code using the example provided in the official documentation, some users may encounter the following error:

AttributeError: can't get attribute 'f' on <module '__main__' (built-in)>

This error stems from a design feature specific to multiprocessing.Pool, where it may encounter difficulties working with functions defined within the main Python script.

Solution:

To resolve this issue, it is necessary to separate the function definition into a separate file and import the module containing the function.

1. Create a new file, e.g., defs.py, and define the function in that file:

<code class="python"># defs.py
def f(x):
    return x*x</code>

2. In the main script, import the module containing the function:

<code class="python"># run.py
from multiprocessing import Pool
import defs

if __name__ == '__main__':
    with Pool(5) as p:
        print(p.map(defs.f, [1, 2, 3]))</code>

Note:

If you had utilized a standard built-in function, such as print, instead of a user-defined function, the original example should have functioned properly. This indicates that the error arises from the design of multiprocessing.Pool in handling custom functions defined within the main script.

以上是使用自訂函數進行多重處理時如何解決 AttributeError？的詳細內容。更多資訊請關注PHP中文網其他相關文章！