Skip to content
Open
Changes from 1 commit
Commits
File filter

Filter by extension

Filter by extension

Conversations
Failed to load comments.
Loading
Jump to
Jump to file
Failed to load files.
Loading
Diff view
Diff view
9 changes: 9 additions & 0 deletions deepmd/main.py
Original file line number Diff line number Diff line change
Expand Up @@ -984,6 +984,15 @@ def main(args: Optional[list[str]] = None) -> None:
RuntimeError
if no command was input
"""
try:
import multiprocessing

# Force fork multiprocessing start method.
multiprocessing.set_start_method("fork", force=True)
logging.info("Successfully set multiprocessing start method to 'fork'.")
except Exception as e:
logging.warning(f"Could not set multiprocessing start method: {e}")

args = parse_args(args=args)

if args.backend not in BACKEND_TABLE:
Expand Down
Loading