fix:更正默认值,避免意料之外的内存占用

用 os.environ.setdefault 将原生库线程数(OMP_NUM_THREADS、OPENBLAS_NUM_THREADS、MKL_NUM_THREADS、NUMEXPR_NUM_THREADS)的默认值设置为 "1"。这样可以确保导入的代码默认不会过度占用 CPU 线程(以及巨量内存)
This commit is contained in:
DawnARC
2026-05-04 17:50:22 +08:00
committed by SengokuCola
parent e962326f93
commit df799cdefa

View File

@@ -0,0 +1,14 @@
from __future__ import annotations
import os
_NATIVE_THREAD_ENV_DEFAULTS = {
"OMP_NUM_THREADS": "1",
"OPENBLAS_NUM_THREADS": "1",
"MKL_NUM_THREADS": "1",
"NUMEXPR_NUM_THREADS": "1",
}
for _name, _value in _NATIVE_THREAD_ENV_DEFAULTS.items():
os.environ.setdefault(_name, _value)