gpt4 book ai didi

python - Numba 矢量化最大化所有处理器

转载 作者:行者123 更新时间:2023-12-01 05:02:08 25 4
gpt4 key购买 nike

有谁知道是否有一种方法可以配置 anaconda,使得 @vectorize 不会占用机器中所有可用的处理器?例如,如果我有一个八核机器,我只希望@vectorize使用四个核。

最佳答案

您可以通过设置 NUMBA_NUM_THREADS 环境变量来限制 target=parallel 将使用的线程数量。请注意,导入 Numba 后您无法更改此设置,它会在您首次启动时设置。您可以通过检查 numba.config.NUMBA_DEFAULT_NUM_THREADS

的值来检查它是否有效

关于python - Numba 矢量化最大化所有处理器,我们在Stack Overflow上找到一个类似的问题: https://stackoverflow.com/questions/25847411/

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