diff --git a/tools/generate_pip_version.py b/tools/generate_pip_version.py
index 519dbfbc5ef..fe034872dfc 100644
--- a/tools/generate_pip_version.py
+++ b/tools/generate_pip_version.py
@@ -13,6 +13,8 @@
 local_label = ""
 version = f"0.8.0"
 
+os.environ["ONEFLOW_RELEASE_VERSION"] = version
+
 # set version if release of nightly
 assert (
     os.getenv("ONEFLOW_RELEASE_VERSION") != ""