From 004952512224512220cc59af2345e2306665535a Mon Sep 17 00:00:00 2001 From: wang <15094922+wcq123123@user.noreply.gitee.com> Date: Thu, 9 Jan 2025 08:50:31 +0000 Subject: [PATCH] =?UTF-8?q?train=5Fperformance=5F1p=5Fdump.sh.=E6=B7=BB?= =?UTF-8?q?=E5=8A=A0=E7=8E=AF=E5=A2=83=E5=8F=98=E9=87=8F=20=E6=B7=BB?= =?UTF-8?q?=E5=8A=A0dockers=E7=9A=84=E7=8E=AF=E5=A2=83=E5=8F=98=E9=87=8F?= MIME-Version: 1.0 Content-Type: text/plain; charset=UTF-8 Content-Transfer-Encoding: 8bit Signed-off-by: wang <15094922+wcq123123@user.noreply.gitee.com> --- .../test/train_performance_1p_dump.sh | 2 ++ 1 file changed, 2 insertions(+) diff --git a/TensorFlow/built-in/recommendation/WideDeep_ID2712_for_TensorFlow/test/train_performance_1p_dump.sh b/TensorFlow/built-in/recommendation/WideDeep_ID2712_for_TensorFlow/test/train_performance_1p_dump.sh index 8c60b984b..f2c6ea7e3 100644 --- a/TensorFlow/built-in/recommendation/WideDeep_ID2712_for_TensorFlow/test/train_performance_1p_dump.sh +++ b/TensorFlow/built-in/recommendation/WideDeep_ID2712_for_TensorFlow/test/train_performance_1p_dump.sh @@ -77,6 +77,8 @@ if [[ $docker_enable == "basic" ]] || [[ $docker_enable == "privileged" ]]; then export LD_LIBRARY_PATH=/usr/local/Ascend/driver/lib64/common/:/usr/local/Ascend/driver/lib64/driver/:$LD_LIBRARY_PATH fi +export PYTHONPATH=/usr/local/Ascend/latest/hccl/python/site-packages/:/usr/local/Ascend/latest/fwkplugin/python/site-packages/:/usr/local/Ascend/latest/compiler/python/site-packages/:/usr/local/Ascend/CANN-7.6/hccl/python/site-packages/:/usr/local/Ascend/CANN-7.6/fwkplugin/python/site-packages/:/usr/local/Ascend/latest/toolkit/python/site-packages/:/usr/local/Ascend/CANN-7.6/compiler/python/site-packages/:/npu/yilai/opensora_plan1.0/decord/python:/npu/yilai/opensora_plan1.0::/usr/local/Ascend/CANN-7.6/opp/built-in/op_impl/ai_core/tbe/:/usr/local/Ascend/latest/opp/built-in/op_impl/ai_core/tbe/:/usr/local/Ascend/latest/opp/built-in/op_impl/ai_core/tbe/:/usr/local/Ascend/latest/opp/built-in/op_impl/ai_core/tbe/:/usr/local/Ascend/latest/opp/built-in/op_impl/ai_core/tbe/:/home/anaconda3/envs/py2/lib/python3.7/site-packages + ##############执行训练########## if [ -d $cur_path/output ];then rm -rf $cur_path/output/* -- Gitee