fix:更正默认值,避免意料之外的内存占用
用 os.environ.setdefault 将原生库线程数(OMP_NUM_THREADS、OPENBLAS_NUM_THREADS、MKL_NUM_THREADS、NUMEXPR_NUM_THREADS)的默认值设置为 "1"。这样可以确保导入的代码默认不会过度占用 CPU 线程(以及巨量内存)
This commit is contained in:
@@ -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)
|
||||||
|
|||||||
Reference in New Issue
Block a user