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() {