fix: fix cuda initialization error in experiment grid (#315)

PKU-Alignment · Apr 9, 2024 · 63bdd34 · 63bdd34
1 parent efe7d9c
commit 63bdd34
Showing 1 changed file with 2 additions and 1 deletion.
diff --git a/omnisafe/common/experiment_grid.py b/omnisafe/common/experiment_grid.py
@@ -17,6 +17,7 @@
 from __future__ import annotations
 
 import json
+import multiprocessing as mp
 import os
 import string
 from concurrent.futures import ProcessPoolExecutor as Pool
@@ -440,7 +441,7 @@ def run(
         announcement = f'\n{joined_var_names}\n\n{line}'
         print(announcement)
 
-        pool = Pool(max_workers=num_pool)
+        pool = Pool(max_workers=num_pool, mp_context=mp.get_context('spawn'))
         # run the variants.
         results = []
         exp_names = []