Mengapa Pemprosesan Berbilang Terhenti pada Teras Tunggal Apabila NumPy Diimport?

Mengapa Gerai Berbilang Pemprosesan pada Teras Tunggal Apabila NumPy Diimport

Apabila cuba menyelaraskan gelung intensif CPU menggunakan joblib, anda mungkin menghadapi isu di mana semua proses pekerja diperuntukkan kepada teras yang sama, yang membawa kepada tiada peningkatan prestasi. Ini berlaku disebabkan oleh modul Python tertentu, termasuk NumPy, memanipulasi pertalian teras pada import.

Punca Punca:

Pautan NumPy terhadap perpustakaan OpenBLAS berbilang benang, yang mengganggu pertalian teras tetapan.

Penyelesaian:

Untuk menyelesaikan isu ini, tetapkan semula perkaitan tugas menggunakan arahan berikut:

<code class="python">os.system("taskset -p 0xff %d" % os.getpid())</code>

Letakkan baris ini selepas mengimport modul di mana isu itu timbul. Ini memaksa proses berjalan pada semua teras yang tersedia.

Penyelesaian Alternatif:

• Pembolehubah Persekitaran: Tetapkan OPENBLAS_MAIN_FREE (atau GOTOBLAS_MAIN_ ) pembolehubah persekitaran kepada 1 pada masa jalan untuk melumpuhkan gelagat penetapan semula afiniti CPU OpenBLAS.
• Pengubahsuaian Masa Binaan: Lumpuhkan tetapan semula pertalian CPU secara kekal dengan mengubah suai Makefile.rule untuk disertakan baris NO_AFFINITY=1 apabila menyusun OpenBLAS daripada sumber.

Atas ialah kandungan terperinci Mengapa Pemprosesan Berbilang Terhenti pada Teras Tunggal Apabila NumPy Diimport?. Untuk maklumat lanjut, sila ikut artikel berkaitan lain di laman web China PHP!