diff --git a/examples/DCNv2/run.sh b/examples/DCNv2/run.sh
index c291d469da5856ffa103d73bd5e75a73856c1d35..b615a77108cbd99a325873872e3823cc11fb6d91 100644
--- a/examples/DCNv2/run.sh
+++ b/examples/DCNv2/run.sh
@@ -49,6 +49,11 @@ export DLRM_CRITEO_DATA_PATH=${dlrm_criteo_data_path}
 export PYTHONPATH=${rec_package_path}:${so_path}:${fore_path}:$PYTHONPATH
 export LD_PRELOAD=/usr/lib64/libgomp.so.1:/usr/lib64/libstdc++.so.6
 export LD_LIBRARY_PATH=${so_path}:/usr/local/lib:$LD_LIBRARY_PATH
+if [ "$(uname -m)" == "aarch64" ]; then
+    export CPLUS_INCLUDE_PATH=/usr/local/gcc7.3.0/include/c++/7.3.0:/usr/local/gcc7.3.0/include/c++/7.3.0/aarch64-unknown-linux-gnu:$CPLUS_INCLUDE_PATH
+else
+    export CPLUS_INCLUDE_PATH=/usr/local/gcc7.3.0/include/c++/7.3.0:/usr/local/gcc7.3.0/include/c++/7.3.0/x86_64-pc-linux-gnu:$CPLUS_INCLUDE_PATH
+fi
 
 export ASCEND_DEVICE_ID=0
 export RANK_ID_START=0