diff --git a/setup.py b/setup.py
index 65ec21e15..f5ff1128f 100644
--- a/setup.py
+++ b/setup.py
@@ -18,6 +18,7 @@ def read_requirements(file_path):
 
 # Determine the package name based on the presence of an environment variable
 package_name = "torchao-nightly" if os.environ.get("TORCHAO_NIGHTLY") else "torchao"
+version_suffix = os.getenv("VERSION_SUFFIX", "")
 
 # Version is year.month.date if using nightlies
 version = current_date if package_name == "torchao-nightly" else "0.2.0"
@@ -42,7 +43,7 @@ def get_extensions():
     if CUDA_HOME is None and torch.cuda.is_available():
         print("CUDA toolkit is not available. Skipping compilation of CUDA extensions")
         print("If you'd like to compile CUDA extensions locally please install the cudatoolkit from https://anaconda.org/nvidia/cuda-toolkit")
-        
+
     use_cuda = torch.cuda.is_available() and CUDA_HOME is not None
     extension = CUDAExtension if use_cuda else CppExtension
 
@@ -85,7 +86,7 @@ def get_extensions():
 
 setup(
     name=package_name,
-    version=version,
+    version=version+version_suffix,
     packages=find_packages(),
     include_package_data=True,
     package_data={