Jupyter Notebook中多进程代码无限运行问题求助

各位好，我最近在Jupyter Notebook里测试一段基础的多进程代码时碰到了麻烦——代码会一直无限运行下去，完全停不下来，但同样的代码放到Spyder里却能正常输出结果，实在摸不着头脑，想问问有没有人知道这是怎么回事？

我的代码如下：

from multiprocessing import Pool

def worker(number):
    return number*number

if __name__ == '__main__':
    pool = Pool() # use all available CPUs
    for square in pool.imap(worker, [0,1,2,3,4]):
        print(square)

备注：内容来源于stack exchange，提问作者Umang Garg