diff --git a/scripts/whl/windows/windows_build_whl.sh b/scripts/whl/windows/windows_build_whl.sh
index d7fd0b8615bca55d8ef98d36f26c591a652778dc..4a958c5aadbeec8ae6b421dff1527b63d7e52e78 100755
--- a/scripts/whl/windows/windows_build_whl.sh
+++ b/scripts/whl/windows/windows_build_whl.sh
@@ -104,6 +104,7 @@ function copy_more_dll() {
     fi
 }
 
+
 function do_build() {
     for ver in ${ALL_PYTHON}
     do
@@ -171,6 +172,7 @@ function do_build() {
         ${PYTHON_DIR}/python3 setup.py bdist_wheel
         cp ${BUILD_DIR}/staging/dist/Meg*.whl ${WINDOWS_WHL_HOME}/
 
+
         echo ""
         echo "##############################################################################################"
         echo "windows whl package location: ${WINDOWS_WHL_HOME}"
@@ -195,5 +197,10 @@ function third_party_prepare() {
 
 ######################
 export ALREADY_CONFIG_PYTHON_VER="yes"
+if [ ${BUILD_WHL_CPU_ONLY} = "OFF" ]; then
+    export SDK_NAME="cu101"
+else
+    export SDK_NAME="cpu"
+fi
 third_party_prepare
 do_build