如何解决在函数内使用多处理
import multiprocessing as mp
def parameters(x,n):
for i in x:
yield (i,n)
def power(a):
x,n = a
return x**n
if __name__ == '__main__':
p = [i for i in range(1000)]
p = parameters(p,2)
agents = 4
chunk = 10
with mp.Pool(processes = agents) as pool:
o = pool.map(power,p,chunksize = chunk)
print(o)
这样我就可以随时调用它。我尝试做这样的事情:
import multiprocessing as mp
def parameters(x,n = a
return x**n
def calculate(s,n):
p = [i for i in range(s)]
p = parameters(p,n)
agents = 4
chunk = 10
with mp.Pool(processes = agents) as pool:
o = pool.map(power,chunksize = chunk)
return o
print(calculate(1000,2))
但是这根本不起作用,它告诉我在一个进程结束之前另一个进程已经开始。但是上面的代码确实有效。有没有办法正确地将此代码放入函数中?如果不是这个设置,那么用什么设置?
解决方法
确保使用 if __name__ == '__main__':
保护只应在主进程中运行的代码。此代码有效:
import multiprocessing as mp
def parameters(x,n):
for i in x:
yield (i,n)
def power(a):
x,n = a
return x**n
def calculate(s,n):
p = [i for i in range(s)]
p = parameters(p,n)
agents = 4
chunk = 10
with mp.Pool(processes = agents) as pool:
o = pool.map(power,p,chunksize = chunk)
return o
if __name__ == '__main__':
print(calculate(1000,2))
如果没有 if
,则会引发以下错误:
RuntimeError:
An attempt has been made to start a new process before the
current process has finished its bootstrapping phase.
This probably means that you are not using fork to start your
child processes and you have forgotten to use the proper idiom
in the main module:
if __name__ == '__main__':
freeze_support()
...
The "freeze_support()" line can be omitted if the program
is not going to be frozen to produce an executable.
版权声明:本文内容由互联网用户自发贡献,该文观点与技术仅代表作者本人。本站仅提供信息存储空间服务,不拥有所有权,不承担相关法律责任。如发现本站有涉嫌侵权/违法违规的内容, 请发送邮件至 dio@foxmail.com 举报,一经查实,本站将立刻删除。