diff --git a/setup.py b/setup.py
index 2f4408a7..603e8a44 100644
--- a/setup.py
+++ b/setup.py
@@ -73,7 +73,7 @@ def get_device_cc():
 
 setup(
     name='bmtrain',
-    version='0.1.7.post1',
+    version='0.1.8',
     author="Guoyang Zeng",
     author_email="qbjooo@qq.com",
     description="A toolkit for training big models",