Default OMP_NUM_THREADS=8 (ultralytics#6770)

fhkiel-mlaip · Feb 24, 2022 · 905c0f1 · 905c0f1
1 parent 814caa9
commit 905c0f1
Showing 1 changed file with 1 addition and 0 deletions.
diff --git a/utils/general.py b/utils/general.py
@@ -45,6 +45,7 @@
 pd.options.display.max_columns = 10
 cv2.setNumThreads(0)  # prevent OpenCV from multithreading (incompatible with PyTorch DataLoader)
 os.environ['NUMEXPR_MAX_THREADS'] = str(NUM_THREADS)  # NumExpr max threads
+os.environ['OMP_NUM_THREADS'] = str(NUM_THREADS)  # OpenMP max threads (PyTorch and SciPy)
 
 
 def is_kaggle():