Rumah  >  Artikel  >  pembangunan bahagian belakang  >  Bagaimana untuk Mengehadkan Penggunaan Benang Numpy Apabila OMP_NUM_THREADS=1 Gagal?

Bagaimana untuk Mengehadkan Penggunaan Benang Numpy Apabila OMP_NUM_THREADS=1 Gagal?

Patricia Arquette
Patricia Arquetteasal
2024-10-19 12:28:02645semak imbas

How to Limit Numpy's Thread Utilization When OMP_NUM_THREADS=1 Fails?

Hadkan Bilangan Benang dalam Numpy

Walaupun percubaan untuk menetapkan OMP_NUM_THREADS=1, dalam situasi tertentu, Numpy mungkin berterusan menggunakan lebih daripada satu utas. Untuk menentukan punca isu ini dan mencapai kiraan benang 1 yang diingini untuk pendaraban matriks, mari kita selidiki pembolehubah persekitaran yang mengawal penggunaan benang.

Konfigurasi Numpy, boleh diperoleh semula melalui numpy.show_config(), mendedahkan penggunaan perpustakaan BLAS (Basic Linear Algebra Subroutines), yang berfungsi sebagai enjin pengiraan. Walau bagaimanapun, mengenal pasti pelaksanaan BLAS khusus yang digunakan oleh Numpy boleh menjadi mencabar.

Untuk mengehadkan bilangan urutan yang digunakan dalam BLAS, kami boleh menetapkan pembolehubah persekitaran sebelum melaksanakan skrip. Pustaka yang biasa digunakan termasuk OpenBLAS, MKL (Intel Math Kernel Library) dan NUMEXPR:

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

Ingat, adalah penting untuk menetapkan pembolehubah ini sebelum menjalankan skrip. Jika berbilang threading masih berterusan, ini mungkin menunjukkan bahawa multithreading diperkenalkan daripada sumber lain, seperti pustaka atau kod tambahan dalam skrip anda.

Atas ialah kandungan terperinci Bagaimana untuk Mengehadkan Penggunaan Benang Numpy Apabila OMP_NUM_THREADS=1 Gagal?. Untuk maklumat lanjut, sila ikut artikel berkaitan lain di laman web China PHP!

Kenyataan:
Kandungan artikel ini disumbangkan secara sukarela oleh netizen, dan hak cipta adalah milik pengarang asal. Laman web ini tidak memikul tanggungjawab undang-undang yang sepadan. Jika anda menemui sebarang kandungan yang disyaki plagiarisme atau pelanggaran, sila hubungi admin@php.cn