HongChuang/ultralytics-ascend
1
0
Fork 0
Code Issues Pull requests Projects Releases Packages Wiki Activity Actions 3

Respect ENV value for OMP_NUM_THREADS if set (#16751)

Browse source 
Co-authored-by: UltralyticsAssistant <web@ultralytics.com>
Co-authored-by: Glenn Jocher <glenn.jocher@ultralytics.com>
This commit is contained in:
Burhan 2024-10-08 14:58:48 -04:00 • committed by GitHub
parent 67683a87f1
commit 1e6c454460
No known key found for this signature in database
GPG key ID: B5690EEEBB952194
1 changed files with 3 additions and 2 deletions
Download patch file Download diff file Expand all files Collapse all files

5
ultralytics/__init__.py
View file

@ -4,8 +4,9 @@ __version__ = "8.3.7"
import os import os
# Set ENV Variables (place before imports) # Set ENV variables (place before imports)
os.environ["OMP_NUM_THREADS"] = "1" # reduce CPU utilization during training if not os.environ.get("OMP_NUM_THREADS"):
os.environ["OMP_NUM_THREADS"] = "1" # default for reduced CPU utilization during training
from ultralytics.data.explorer.explorer import Explorer from ultralytics.data.explorer.explorer import Explorer
from ultralytics.models import NAS, RTDETR, SAM, YOLO, FastSAM, YOLOWorld from ultralytics.models import NAS, RTDETR, SAM, YOLO, FastSAM, YOLOWorld