如何限制 NumPy 中的執行緒使用以獲得最佳效能？

自訂 NumPy 中的執行緒使用

NumPy 使用多個執行緒進行各種操作，這可能會影響多核心系統的效能。為了限制 NumPy 使用的執行緒數量，可以設定幾個環境變數。

NumPy 中控制線程使用的環境變數可能會有所不同，具體取決於所使用的底層線性代數庫，例如 BLAS 。若要識別正在使用的函式庫，您可以使用 numpy.show_config() 函數。

如果您的Numpy 配置顯示它正在使用BLAS，您可以在執行腳本之前嘗試設定以下環境變數：

• export MKL_NUM_THREADS=1
• export MKL_NUM_THREADS=1
• export MKL_NUM_THREADS=1

export MKLcEA_MNUA_DPTA​​DPNUM&A_F​​D. =1

export OMP_NUM_THREADS=1

這些設定應限制用於的執行緒數NumPy 中的矩陣乘法和其他計算。 請記住，確定程式碼中多執行緒的確切來源可能具有挑戰性。如果上述變數無法解決問題，則值得探索可能影響線程行為的其他特定於庫的環境標誌。

以上是如何限制 NumPy 中的執行緒使用以獲得最佳效能？的詳細內容。更多資訊請關注PHP中文網其他相關文章！