如何限制 NumPy 中的线程使用以获得最佳性能？

自定义 NumPy 中的线程使用

NumPy 使用多个线程进行各种操作，这可能会影响多核系统的性能。为了限制 NumPy 使用的线程数量，可以设置几个环境变量。

NumPy 中控制线程使用的环境变量可能会有所不同，具体取决于所使用的底层线性代数库，例如 BLAS 。要识别正在使用的库，您可以使用 numpy.show_config() 函数。

如果您的 Numpy 配置显示它正在使用 BLAS，您可以在执行脚本之前尝试设置以下环境变量：

• export MKL_NUM_THREADS=1
• export NUMEXPR_NUM_THREADS=1
• export OMP_NUM_THREADS=1

这些设置应限制用于的线程数NumPy 中的矩阵乘法和其他计算。

请记住，确定代码中多线程的确切来源可能具有挑战性。如果上述变量不能解决问题，则值得探索可能影响线程行为的其他特定于库的环境标志。

以上是如何限制 NumPy 中的线程使用以获得最佳性能？的详细内容。更多信息请关注PHP中文网其他相关文章！