diff --git a/utils/datasets.py b/utils/datasets.py
index 3d724e6fae62..2da3940c3c95 100755
--- a/utils/datasets.py
+++ b/utils/datasets.py
@@ -59,7 +59,7 @@ def create_dataloader(path, imgsz, batch_size, stride, opt, hyp=None, augment=Fa
                                     pad=pad)
 
     batch_size = min(batch_size, len(dataset))
-    nw = min([os.cpu_count()//opt.world_size, batch_size if batch_size > 1 else 0, 8])  # number of workers
+    nw = min([os.cpu_count()//(opt.world_size if hasattr(opt, "world_size") else 1), batch_size if batch_size > 1 else 0, 8])  # number of workers
     train_sampler = torch.utils.data.distributed.DistributedSampler(dataset) if local_rank != -1 else None
     dataloader = torch.utils.data.DataLoader(dataset,
                                              batch_size=batch_size,