From cd40f843dbe3f238d440469e202430ff9716b152 Mon Sep 17 00:00:00 2001
From: Shuming Liu <smshine@qq.com>
Date: Sat, 31 Jan 2026 16:31:50 +0800
Subject: [PATCH] fixed a bug

---
 video_cleaner.py | 7 +++++++
 1 file changed, 7 insertions(+)

diff --git a/video_cleaner.py b/video_cleaner.py
index 394cc1c..82172ac 100644
--- a/video_cleaner.py
+++ b/video_cleaner.py
@@ -166,6 +166,13 @@ class VideoCleaner:
 
 if __name__ == "__main__":
     import argparse
+    # 设置多进程启动方法为 'spawn'，这在 CUDA 环境下更安全，能避免显存冲突
+    import multiprocessing
+    try:
+        multiprocessing.set_start_method('spawn', force=True)
+    except RuntimeError:
+        pass
+
     parser = argparse.ArgumentParser()
     parser.add_argument("dir", help="Target directory")
     parser.add_argument("--days", type=int, default=30)