Python 代碼使用多進程無限運行

Question

我正在嘗試使用多處理在 jupyter notebook 中執行以下代碼，但循環無限運行。
我需要幫助解決這個問題。

import multiprocessing as mp
import numpy as np

def square(x):
    return np.square(x)

x = np.arange(64)

pool = mp.Pool(4)
squared = pool.map(square, [x[16*i:16*i+16] for i in range(4)])

mp.cpu_count()的 output 是 4。

Answer 1

您需要將代碼重寫為：

def main():
    x = np.arange(64)
    pool = mp.Pool(4)
    squared = .....

if __name__ == '__main__':
    main()

此代碼當前正在每個進程中運行。 您需要它只在執行設置的一個進程中運行。

Answer 2

你忘了：

pool.close()
pool.join()