diff --git a/setup.py b/setup.py
index 7b94e6b33744bc2bd467af775240548341c4db4d..57b41d393f5da4f0c52cec6094f533671edf9f23 100644
--- a/setup.py
+++ b/setup.py
@@ -11,7 +11,7 @@ def get_version():
     # 读取文件内容，使用正则找出版本号
     with open(init_path, "r", encoding="utf-8") as f:
         content = f.read()
-        version_match = re.search(r'^__version__ = ["\']([^"\']*)["\']', content, re.M)
+        version_match = re.search(r'^__version__ ?= ?["\']([^"\']*)["\']', content, re.M)
         if version_match:
             return version_match.group(1)
         raise RuntimeError("无法在 __init__.py 中找到 __version__ 定义")