set multiprocessing start method to spawn

Signed-off-by: Jinzhe Zeng <[email protected]>
deepmodeling · Jan 27, 2024 · 956f16c · 956f16c
1 parent c62932f
commit 956f16c
Showing 1 changed file with 1 addition and 0 deletions.
diff --git a/source/tests/pt/__init__.py b/source/tests/pt/__init__.py
@@ -3,3 +3,4 @@
 
 torch.set_num_threads(1)
 torch.set_num_interop_threads(1)
+torch.multiprocessing.set_start_method("spawn")