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

时间:2014-09-15 11:49:52

标签: python vectorization numba

有没有人知道是否有办法配置anaconda,以便@vectorize不会占用机器中的所有处理器?例如,如果我有一台八核机器,我只希望@vectorize使用四个核心。

1 个答案:

答案 0 :(得分:2)

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

的值来检查它是否有效