OSSForks/faceswap
1
0
Fork 0
mirror of https://github.com/zebrajr/faceswap.git synced 2025-12-06 00:20:09 +01:00
Code Issues Actions Packages Projects Releases Wiki Activity

bugfix - numexpr. ignore OMP_NUM_THREADS EnvVar

Browse Source
This commit is contained in:
torzdf 2023-01-18 15:39:05 +00:00
parent e2d84bec50
commit 1d19aea281
1 changed files with 4 additions and 0 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

4
lib/cli/launcher.py
View File

@ -57,6 +57,10 @@ class ScriptExecutor(): # pylint:disable=too-few-public-methods
# Allocate a decent number of threads to numexpr to suppress warnings
cpu_count = os.cpu_count()
allocate = cpu_count - cpu_count // 3 if cpu_count is not None else 1
if "OMP_NUM_THREADS" in os.environ:
# If this is set above NUMEXPR_MAX_THREADS, numexpr will error.
# ref: https://github.com/pydata/numexpr/issues/322
os.environ.pop("OMP_NUM_THREADS")
os.environ["NUMEXPR_MAX_THREADS"] = str(max(1, allocate))
# Ensure tensorflow doesn't pin all threads to one core when using Math Kernel Library