gpt4 book ai didi

python - Python 中意外的多线程 (scikit-learn)

转载 作者:太空狗 更新时间:2023-10-30 00:36:20 25 4
gpt4 key购买 nike

我正在为高斯混合模型使用 sklearn 模块的混合子模块...当我在多核系统上运行我的代码时,即使我没有在代码中要求它,它也会使用多个内核。这是默认行为吗?更重要的是,我该如何禁用它?

谢谢

最佳答案

如果您使用的是 MKL,请尝试

导出 MKL_NUM_THREADS=1

对于带有 OpenBLAS 的 Numpy:

导出 OPENBLAS_NUM_THREADS=1

对于某些版本的 Numpy,建议使用这种变体:

导出 NUMEXPR_NUM_THREADS=1

必须在脚本运行之前设置环境变量(在脚本内部设置不会产生预期的效果)。要在运行时设置线程,请参阅:Set max number of threads at runtime on numpy/openblas

请参阅以下内容以确定您的 numpy 是如何设置的:How to check blas/lapack linkage in numpy/scipy?

关于python - Python 中意外的多线程 (scikit-learn),我们在Stack Overflow上找到一个类似的问题: https://stackoverflow.com/questions/19257070/

25 4 0
Copyright 2021 - 2024 cfsdn All Rights Reserved 蜀ICP备2022000587号
广告合作:1813099741@qq.com 6ren.com