diff --git a/ultralytics/__init__.py b/ultralytics/__init__.py index bb2ea3f62e..4c98c5c08f 100644 --- a/ultralytics/__init__.py +++ b/ultralytics/__init__.py @@ -1,6 +1,6 @@ # Ultralytics YOLO 🚀, AGPL-3.0 license -__version__ = "8.2.7" +__version__ = "8.2.8" from ultralytics.data.explorer.explorer import Explorer from ultralytics.models import RTDETR, SAM, YOLO, YOLOWorld diff --git a/ultralytics/data/build.py b/ultralytics/data/build.py index d0773737a2..6fa611cc40 100644 --- a/ultralytics/data/build.py +++ b/ultralytics/data/build.py @@ -20,7 +20,7 @@ from ultralytics.data.loaders import ( autocast_list, ) from ultralytics.data.utils import IMG_FORMATS, VID_FORMATS -from ultralytics.utils import NUM_THREADS, RANK, colorstr +from ultralytics.utils import LINUX, NUM_THREADS, RANK, colorstr from ultralytics.utils.checks import check_file from .dataset import GroundingDataset, YOLODataset, YOLOMultiModalDataset @@ -81,7 +81,8 @@ def seed_worker(worker_id): # noqa worker_seed = torch.initial_seed() % 2**32 np.random.seed(worker_seed) random.seed(worker_seed) - os.sched_setaffinity(0, range(NUM_THREADS)) # fix https://github.com/ultralytics/ultralytics/pull/11195 + if LINUX and hasattr(os, "sched_setaffinity"): # unsupported on macOS and Windows + os.sched_setaffinity(0, range(NUM_THREADS)) # fix https://github.com/ultralytics/ultralytics/pull/11195 def build_yolo_dataset(cfg, img_path, batch, data, mode="train", rect=False, stride=32, multi_modal=False):