diff --git a/research/cv/Neighbor2Neighbor/scripts/run_infer_310.sh b/research/cv/Neighbor2Neighbor/scripts/run_infer_310.sh
index 136c7775350a49f50f20e25848464b0ea572751c..e0c495aa8f21235f62cae6211927ed3cd0c1d98f 100644
--- a/research/cv/Neighbor2Neighbor/scripts/run_infer_310.sh
+++ b/research/cv/Neighbor2Neighbor/scripts/run_infer_310.sh
@@ -39,19 +39,19 @@ echo "dataset: "$dataset
 echo "input_path: "$input_path
 echo "device id: "$device_id
 
-export ASCEND_HOME=/usr/local/Ascend
-
-export PATH=$ASCEND_HOME/fwkacllib/ccec_compiler/bin:$ASCEND_HOME/fwkacllib/bin:$ASCEND_HOME/toolkit/bin:$PATH
-
-export LD_LIBRARY_PATH=/usr/local/lib/:/usr/local/fwkacllib/lib64:$ASCEND_HOME/driver/lib64:$ASCEND_HOME/add-ons:/usr/local/Ascend/toolkit/lib64:$LD_LIBRARY_PATH
-
-export PYTHONPATH=$ASCEND_HOME/fwkacllib/python/site-packages
-
-export PATH=/usr/local/python375/bin:$PATH
-export NPU_HOST_LIB=/usr/local/Ascend/acllib/lib64/stub
-export ASCEND_OPP_PATH=/usr/local/Ascend/opp
-export ASCEND_AICPU_PATH=/usr/local/Ascend
-export LD_LIBRARY_PATH=/usr/local/lib64/:$LD_LIBRARY_PATH
+export ASCEND_HOME=/usr/local/Ascend/
+if [ -d ${ASCEND_HOME}/ascend-toolkit ]; then
+    export PATH=$ASCEND_HOME/fwkacllib/bin:$ASCEND_HOME/fwkacllib/ccec_compiler/bin:$ASCEND_HOME/ascend-toolkit/latest/fwkacllib/ccec_compiler/bin:$ASCEND_HOME/ascend-toolkit/latest/atc/bin:$PATH
+    export LD_LIBRARY_PATH=$ASCEND_HOME/fwkacllib/lib64:/usr/local/lib:$ASCEND_HOME/ascend-toolkit/latest/atc/lib64:$ASCEND_HOME/ascend-toolkit/latest/fwkacllib/lib64:$ASCEND_HOME/driver/lib64:$ASCEND_HOME/add-ons:$LD_LIBRARY_PATH
+    export TBE_IMPL_PATH=$ASCEND_HOME/ascend-toolkit/latest/opp/op_impl/built-in/ai_core/tbe
+    export PYTHONPATH=$ASCEND_HOME/fwkacllib/python/site-packages:${TBE_IMPL_PATH}:$ASCEND_HOME/ascend-toolkit/latest/fwkacllib/python/site-packages:$PYTHONPATH
+    export ASCEND_OPP_PATH=$ASCEND_HOME/ascend-toolkit/latest/opp
+else
+    export PATH=$ASCEND_HOME/fwkacllib/bin:$ASCEND_HOME/fwkacllib/ccec_compiler/bin:$ASCEND_HOME/atc/ccec_compiler/bin:$ASCEND_HOME/atc/bin:$PATH
+    export LD_LIBRARY_PATH=$ASCEND_HOME/fwkacllib/lib64:/usr/local/lib:$ASCEND_HOME/atc/lib64:$ASCEND_HOME/acllib/lib64:$ASCEND_HOME/driver/lib64:$ASCEND_HOME/add-ons:$LD_LIBRARY_PATH
+    export PYTHONPATH=$ASCEND_HOME/fwkacllib/python/site-packages:$ASCEND_HOME/atc/python/site-packages:$PYTHONPATH
+    export ASCEND_OPP_PATH=$ASCEND_HOME/opp
+fi
 
 function compile_app()
 {