diff --git a/PyTorch/contrib/cv/classification/InceptionV4_ID1778_for_PyTorch/requirements.txt b/PyTorch/contrib/cv/classification/InceptionV4_ID1778_for_PyTorch/requirements.txt index 9ec5e11346420a23cb950967bbfa73d5da2d52b9..eb42c5baa28ea370166c4de7939714e78893b78f 100644 --- a/PyTorch/contrib/cv/classification/InceptionV4_ID1778_for_PyTorch/requirements.txt +++ b/PyTorch/contrib/cv/classification/InceptionV4_ID1778_for_PyTorch/requirements.txt @@ -1,5 +1,7 @@ -torchvision +torchvision==0.9.1 munch tqdm scipy -pillow==9.1.0 \ No newline at end of file +pillow==9.1.0 +sympy +decorator \ No newline at end of file diff --git a/PyTorch/contrib/cv/classification/InceptionV4_ID1778_for_PyTorch/test/env_npu.sh b/PyTorch/contrib/cv/classification/InceptionV4_ID1778_for_PyTorch/test/env_npu.sh index c3c1ea294f381c983e6ffb1b4f56b7579e96b4cb..e2ce4cfad237bd7293ebf6cb5efccdd8762110db 100644 --- a/PyTorch/contrib/cv/classification/InceptionV4_ID1778_for_PyTorch/test/env_npu.sh +++ b/PyTorch/contrib/cv/classification/InceptionV4_ID1778_for_PyTorch/test/env_npu.sh @@ -1,27 +1,13 @@ #!/bin/bash -# 配置CANN相关环境变量 -CANN_INSTALL_PATH_CONF='/etc/Ascend/ascend_cann_install.info' - -if [ -f $CANN_INSTALL_PATH_CONF ]; then - CANN_INSTALL_PATH=$(cat $CANN_INSTALL_PATH_CONF | grep Install_Path | cut -d "=" -f 2) -else - CANN_INSTALL_PATH="/usr/local/Ascend/" -fi - -if [ -d ${CANN_INSTALL_PATH}/ascend-toolkit/latest ];then - source ${CANN_INSTALL_PATH}/ascend-toolkit/set_env.sh -else - source ${CANN_INSTALL_PATH}/nnae/set_env.sh -fi - -msnpureport -g error -d 0 -msnpureport -g error -d 1 -msnpureport -g error -d 2 -msnpureport -g error -d 3 -msnpureport -g error -d 4 -msnpureport -g error -d 5 -msnpureport -g error -d 6 -msnpureport -g error -d 7 +export install_path=/usr/local/Ascend + +export LD_LIBRARY_PATH=/usr/local/Ascend/ascend-toolkit/latest/lib64:/usr/local/Ascend/ascend-toolkit/latest/lib64/plugin/opskernel:/usr/local/Ascend/ascend-toolkit/latest/lib64/plugin/nnengine:$LD_LIBRARY_PATH +export PYTHONPATH=/usr/local/Ascend/ascend-toolkit/latest/python/site-packages:/usr/local/Ascend/ascend-toolkit/latest/opp/op_impl/built-in/ai_core/tbe:$PYTHONPATH +export PATH=/usr/local/Ascend/ascend-toolkit/latest/bin:/usr/local/Ascend/ascend-toolkit/latest/compiler/ccec_compiler/bin:$PATH +export ASCEND_AICPU_PATH=/usr/local/Ascend/ascend-toolkit/latest +export ASCEND_OPP_PATH=/usr/local/Ascend/ascend-toolkit/latest/opp +export TOOLCHAIN_HOME=/usr/local/Ascend/ascend-toolkit/latest/toolkit +export ASCEND_HOME_PATH=/usr/local/Ascend/ascend-toolkit/latest #将Host日志输出到串口,0-关闭/1-开启 export ASCEND_SLOG_PRINT_TO_STDOUT=0 @@ -34,12 +20,20 @@ export TASK_QUEUE_ENABLE=1 #设置是否开启PTCopy,0-关闭/1-开启 export PTCOPY_ENABLE=1 #设置是否开启combined标志,0-关闭/1-开启 -export COMBINED_ENABLE=1 -#设置特殊场景是否需要重新编译,不需要修改 -export DYNAMIC_OP="ADD#MUL" #HCCL白名单开关,1-关闭/0-开启 export HCCL_WHITELIST_DISABLE=1 +#设置device侧日志登记为error +${install_path}/driver/tools/msnpureport -g error -d 0 +${install_path}/driver/tools/msnpureport -g error -d 1 +${install_path}/driver/tools/msnpureport -g error -d 2 +${install_path}/driver/tools/msnpureport -g error -d 3 +${install_path}/driver/tools/msnpureport -g error -d 4 +${install_path}/driver/tools/msnpureport -g error -d 5 +${install_path}/driver/tools/msnpureport -g error -d 6 +${install_path}/driver/tools/msnpureport -g error -d 7 +#关闭Device侧Event日志 +${install_path}/driver/tools/msnpureport -e disable path_lib=$(python3.7 -c """ import sys