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 8c60b984bd653fc53fb6ec1230fcdefee7fc6885..f2c6ea7e35d89af16b489d1b5d57bc86b7ac9b82 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/*