我正在为高斯混合模型使用 sklearn 模块的混合子模块......当我在多核系统上运行我的代码时,它使用多个内核,即使我没有在代码中要求它。这是默认行为吗?更重要的是,我怎样才能禁用它?
谢谢
我正在为高斯混合模型使用 sklearn 模块的混合子模块......当我在多核系统上运行我的代码时,它使用多个内核,即使我没有在代码中要求它。这是默认行为吗?更重要的是,我怎样才能禁用它?
谢谢
如果您使用的是 MKL,请尝试
export MKL_NUM_THREADS=1
对于带有 OpenBLAS 的 Numpy:
export OPENBLAS_NUM_THREADS=1
对于 Numpy 的某些版本,建议使用此变体:
export NUMEXPR_NUM_THREADS=1
必须在脚本运行之前设置环境变量(在脚本本身内部设置不会产生预期的效果)。有关在运行时设置线程,请参阅:Set max number of threads at runtime on numpy/openblas
请参阅以下内容以确定您的 numpy 是如何设置的:如何检查 numpy/scipy 中的 blas/lapack 链接?