diff --git a/onnx/cppbuild.sh b/onnx/cppbuild.sh
index 991a90e113..e4aff2870b 100755
--- a/onnx/cppbuild.sh
+++ b/onnx/cppbuild.sh
@@ -107,7 +107,7 @@ sedinplace 's/const std::string /std::string /g' onnx/defs/schema.h
 sedinplace 's/if WINDOWS:/if False:/g' setup.py
 sedinplace '/if platform.architecture/{N;N;N;d;}' setup.py
 sedinplace "/setup_requires.append('pytest-runner')/d" setup.py
-export CMAKE_ARGS="-DCMAKE_INSTALL_PREFIX=$INSTALL_PATH -DONNX_USE_PROTOBUF_SHARED_LIBS=OFF -DProtobuf_USE_STATIC_LIBS=ON -DONNX_USE_LITE_PROTO=ON"
+export CMAKE_ARGS="-DCMAKE_INSTALL_PREFIX=$INSTALL_PATH -DONNX_USE_PROTOBUF_SHARED_LIBS=OFF -DProtobuf_USE_STATIC_LIBS=ON -DONNX_USE_LITE_PROTO=ON -DPYTHON_LIBRARY=$PYTHON_LIB_PATH -DPYTHON_INCLUDE_DIR=$PYTHON_INCLUDE_PATH"
 $PYTHON_BIN_PATH -m pip install --target=$PYTHON_LIB_PATH setuptools==67.6.1 protobuf==$PROTO
 $PYTHON_BIN_PATH -m pip install -e .