diff --git a/build.sh b/build.sh
index 05db6e192a405c0b46ad097008b637d7fa530a2c..f7cb9fe410b8ff5ea0dd2a5c9e1d2ff83f952ddb 100644
--- a/build.sh
+++ b/build.sh
@@ -27,6 +27,8 @@ for pybin in "${PYTHON_BINS[@]}"; do
     export BUILD_WHEEL_FLAGS=
     export CFLAGS="-Wno-maybe-uninitialized -Wno-uninitialized -Wno-free-nonheap-object -Wno-restrict"
     export CXXFLAGS=$CFLAGS
+    export PYTORCH_BUILD_VERSION=2.6.0
+    export PYTORCH_BUILD_NUMBER=0
     #export CUDA_C_FLAGS=
     # ================================
     build_wheel $pybin