Attempt to add multiprocessing support for Windows users

vortex-exoplanet · Jul 31, 2024 · 2ed8e35 · 2ed8e35
1 parent bf89d41
commit 2ed8e35
Showing 1 changed file with 7 additions and 1 deletion.
diff --git a/vip_hci/config/utils_conf.py b/vip_hci/config/utils_conf.py
@@ -471,7 +471,13 @@ def pool_map(nproc, fkt, *args, **kwargs):
         if not _generator:
             res = list(res)
     else:
-        multiprocessing.set_start_method("fork", force=True)
+        # Check available start methods and pick accordingly (machine-dependent)
+        avail_methods = multiprocessing.get_all_start_methods()
+        if 'fork' in avail_methods:
+            # faster when available
+            multiprocessing.set_start_method("fork", force=True)
+        else:
+            multiprocessing.set_start_method("spawn", force=True)
         from multiprocessing import Pool
 
         # deactivate multithreading