ntsc207 commited on
Commit
825c5d1
·
verified ·
1 Parent(s): 9ad66ab

Change limit cpus 1 to 8

Browse files
Files changed (1) hide show
  1. detect_strongsort.py +5 -5
detect_strongsort.py CHANGED
@@ -2,11 +2,11 @@ import argparse
2
 
3
  import os
4
  # limit the number of cpus used by high performance libraries
5
- os.environ["OMP_NUM_THREADS"] = "1"
6
- os.environ["OPENBLAS_NUM_THREADS"] = "1"
7
- os.environ["MKL_NUM_THREADS"] = "1"
8
- os.environ["VECLIB_MAXIMUM_THREADS"] = "1"
9
- os.environ["NUMEXPR_NUM_THREADS"] = "1"
10
  import platform
11
  import sys
12
  import numpy as np
 
2
 
3
  import os
4
  # limit the number of cpus used by high performance libraries
5
+ os.environ["OMP_NUM_THREADS"] = "8"
6
+ os.environ["OPENBLAS_NUM_THREADS"] = "8"
7
+ os.environ["MKL_NUM_THREADS"] = "8"
8
+ os.environ["VECLIB_MAXIMUM_THREADS"] = "8"
9
+ os.environ["NUMEXPR_NUM_THREADS"] = "8"
10
  import platform
11
  import sys
12
  import numpy as np