set USE_LIBUV=0 on windows

Lightning-AI · Nov 8, 2024 · aa213b6 · aa213b6
1 parent c8ffd2e
commit aa213b6
Showing 1 changed file with 5 additions and 0 deletions.
diff --git a/src/lightning/fabric/__init__.py b/src/lightning/fabric/__init__.py
@@ -2,6 +2,7 @@
 
 import logging
 import os
+import sys
 
 from lightning_utilities.core.imports import package_available
 
@@ -26,6 +27,10 @@
 # https://github.com/pytorch/pytorch/issues/83973
 os.environ["PYTORCH_NVML_BASED_CUDA_CHECK"] = "1"
 
+# see https://github.com/pytorch/pytorch/issues/139990
+if sys.platform == "win32":
+    os.environ["USE_LIBUV"] = "0"
+
 
 from lightning.fabric.fabric import Fabric  # noqa: E402
 from lightning.fabric.utilities.seed import seed_everything  # noqa: E402