Sklearn在多个函数中使用n_jobs参数。我们可以传递我们想要使用多少个核心。例如,一个核心为一个,所有核心为-1个。我的系统有六个核心,我只想使用0,1,3,4核。这个是可能的吗?
n_jobs
发布于 2021-08-08 13:37:46
我从来没有尝试过,也没有听说过这种可能性,用scikit-learn。
scikit-learn
不过,作为scikit-learn 使用 joblib,您应该开始研究如何实现自定义ParallelBackendBase,这是一个助手abc,它定义了ParallelBackend必须实现的所有方法1、2、3.;您可能能够在内部实现CPU亲和力,将进程固定在特定的CPU内核上。
joblib
ParallelBackendBase
ParallelBackend
不过,这纯粹是实验性的。
发布于 2021-08-08 11:56:50
不,这是不可能的:您不能指定特定的核心,这取决于系统。
https://stackoverflow.com/questions/68700344
相似问题