diff --git a/setup.py b/setup.py
index 3c4cf6f4..7c29a10f 100644
--- a/setup.py
+++ b/setup.py
@@ -73,7 +73,7 @@ def get_device_cc():
 
 setup(
     name='bmtrain',
-    version='0.2.1',
+    version='0.2.2',
     author="Guoyang Zeng",
     author_email="qbjooo@qq.com",
     description="A toolkit for training big models",