Fix weird docker error (#2581)

(cherry picked from commit b0da59e)

src/torchmetrics/metric.py

@@ -107,7 +107,8 @@ def __init__(
         # see (https://github.com/pytorch/pytorch/blob/3e6bb5233f9ca2c5aa55d9cda22a7ee85439aa6e/
         # torch/nn/modules/module.py#L227)
         torch._C._log_api_usage_once(f"torchmetrics.metric.{self.__class__.__name__}")
-
+        # magic patch for `RuntimeError: DataLoader worker (pid(s) 104) exited unexpectedly`
+        self._TORCH_GREATER_EQUAL_2_1 = bool(_TORCH_GREATER_EQUAL_2_1)
         self._device = torch.device("cpu")
         self._dtype = torch.get_default_dtype()
 
@@ -441,7 +442,7 @@ def _sync_dist(self, dist_sync_fn: Callable = gather_all_tensors, process_group:
 
             # cornor case in distributed settings where a rank have not received any data, create empty to concatenate
             if (
-                _TORCH_GREATER_EQUAL_2_1
+                self._TORCH_GREATER_EQUAL_2_1
                 and reduction_fn == dim_zero_cat
                 and isinstance(input_dict[attr], list)
                 and len(input_dict[attr]) == 0