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

Question

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

Answer 1

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

的值来检查它是否有效