diff --git a/.github/workflows/release.yml b/.github/workflows/release.yml
index 76c3478d..79d416e2 100644
--- a/.github/workflows/release.yml
+++ b/.github/workflows/release.yml
@@ -22,7 +22,8 @@ jobs:
         python release/check_release_version.py --release_version ${{ env.RELEASE_VERSION }}
     - name: Build DeepSpeed-MII
       run: |
-        MII_BUILD_STRING=" " python setup.py bdist_wheel
+        pip install build
+        MII_BUILD_STRING=" " python -m build --wheel
     - name: Publish to PyPI
       uses: pypa/gh-action-pypi-publish@release/v1
       with:
diff --git a/release/release.sh b/release/release.sh
index dc5f0329..e88e2535 100644
--- a/release/release.sh
+++ b/release/release.sh
@@ -35,7 +35,7 @@ if [ $? != 0 ]; then
     exit 1
 fi
 
-MII_BUILD_STRING="" python setup.py bdist_wheel
+MII_BUILD_STRING="" python -m build --wheel
 WHL=deepspeed_mii-${version}-py3-none-any.whl
 
 if [ ! -f dist/${WHL} ]; then
diff --git a/setup.py b/setup.py
index aecebcf6..6c594792 100644
--- a/setup.py
+++ b/setup.py
@@ -47,7 +47,7 @@ def command_exists(cmd):
 version_str = open('version.txt', 'r').read().strip()
 
 # Build specifiers like .devX can be added at install time. Otherwise, add the git hash.
-# example: MII_BUILD_STR=".dev20201022" python setup.py sdist bdist_wheel
+# example: MII_BUILD_STR=".dev20201022" python -m build --sdist --wheel
 
 MII_BUILD_STRING = 'MII_BUILD_STRING'
 BUILD_FILE = 'build.txt'