Scikit-learn 機器學習模型訓練使用多個 CPU

Question

我想通過使用高端 EC2 實例來減少模型的訓練時間。 因此，我嘗試了具有 2 個 CPU 的c5.18xlarge實例並使用參數n_jobs=-1運行了幾個模型，但我注意到只使用了一個 CPU：

我可以以某種方式讓 Scikit-learn 使用所有 CPU 嗎？

Answer 1

嘗試添加：

import multiprocessing
multiprocessing.set_start_method('forkserver')

在代碼的頂部，在運行或導入任何東西之前。 這是python中多處理的一個眾所周知的問題。