diff --git a/TensorFlow/built-in/cv/detection/SSD-Resnet34_ID0048_for_TensorFlow/test/set_conda.sh b/TensorFlow/built-in/cv/detection/SSD-Resnet34_ID0048_for_TensorFlow/test/set_conda.sh new file mode 100644 index 0000000000000000000000000000000000000000..febb0fa34937160c9ad16c92dcb67cd4005e2f57 --- /dev/null +++ b/TensorFlow/built-in/cv/detection/SSD-Resnet34_ID0048_for_TensorFlow/test/set_conda.sh @@ -0,0 +1,2 @@ +export PATH=/home/anaconda3/bin:$PATH +export LD_LIBRARY_PATH=/home/anaconda3/lib:$LD_LIBRARY_PATH \ No newline at end of file diff --git a/TensorFlow/built-in/cv/detection/SSD-Resnet34_ID0048_for_TensorFlow/test/train_performance_1p.sh b/TensorFlow/built-in/cv/detection/SSD-Resnet34_ID0048_for_TensorFlow/test/train_performance_1p.sh index 5a9d324db6e368150784973d889389fa72bb177b..deffce9053d5174019d704a37a692d197180c636 100644 --- a/TensorFlow/built-in/cv/detection/SSD-Resnet34_ID0048_for_TensorFlow/test/train_performance_1p.sh +++ b/TensorFlow/built-in/cv/detection/SSD-Resnet34_ID0048_for_TensorFlow/test/train_performance_1p.sh @@ -88,6 +88,11 @@ do cp -rf $install_path/fwkacllib/data/rl/Ascend910/custom ${autotune_dump_path}/RL/ elif [[ $para == --data_path* ]];then data_path=`echo ${para#*=}` + elif [[ $para == --conda_name* ]];then + conda_name=`echo ${para#*=}` + source set_conda.sh + echo "conda_name:$conda_name" + source activate $conda_name fi done diff --git a/TensorFlow/built-in/cv/detection/SSD-Resnet34_ID0048_for_TensorFlow/test/train_performance_8p.sh b/TensorFlow/built-in/cv/detection/SSD-Resnet34_ID0048_for_TensorFlow/test/train_performance_8p.sh index b0eeca57ba7f15094c0cbef0a24870111aa802fb..71b0e65c0c1a5dec5147e6c9a571e1f6ee316cf1 100644 --- a/TensorFlow/built-in/cv/detection/SSD-Resnet34_ID0048_for_TensorFlow/test/train_performance_8p.sh +++ b/TensorFlow/built-in/cv/detection/SSD-Resnet34_ID0048_for_TensorFlow/test/train_performance_8p.sh @@ -89,6 +89,11 @@ do elif [[ $para == --bind_core* ]]; then bind_core=`echo ${para#*=}` name_bind="_bindcore" + elif [[ $para == --conda_name* ]];then + conda_name=`echo ${para#*=}` + source set_conda.sh + echo "conda_name:$conda_name" + source activate $conda_name fi done diff --git a/TensorFlow/built-in/cv/detection/YoloV3_ID0076_for_TensorFlow/test/train_performance_1p.sh b/TensorFlow/built-in/cv/detection/YoloV3_ID0076_for_TensorFlow/test/train_performance_1p.sh index 604af32e6d1e3969b18d06db8ca72890739cd1ec..6e0163f1ecdd0b4de1c094a444c51a91ebe0a3eb 100644 --- a/TensorFlow/built-in/cv/detection/YoloV3_ID0076_for_TensorFlow/test/train_performance_1p.sh +++ b/TensorFlow/built-in/cv/detection/YoloV3_ID0076_for_TensorFlow/test/train_performance_1p.sh @@ -86,6 +86,11 @@ do cp -rf $install_path/fwkacllib/data/rl/Ascend910/custom ${autotune_dump_path}/RL/ elif [[ $para == --data_path* ]];then data_path=`echo ${para#*=}` + elif [[ $para == --conda_name* ]];then + conda_name=`echo ${para#*=}` + source set_conda.sh + echo "conda_name:$conda_name" + source activate $conda_name fi done diff --git a/TensorFlow/built-in/cv/detection/YoloV3_ID0076_for_TensorFlow/test/train_performance_8p.sh b/TensorFlow/built-in/cv/detection/YoloV3_ID0076_for_TensorFlow/test/train_performance_8p.sh index 36704398cf9b8fd93fd8fc29c4358303585d2a31..fa221fea271a4579a7043c888cab13dbed45088f 100644 --- a/TensorFlow/built-in/cv/detection/YoloV3_ID0076_for_TensorFlow/test/train_performance_8p.sh +++ b/TensorFlow/built-in/cv/detection/YoloV3_ID0076_for_TensorFlow/test/train_performance_8p.sh @@ -85,6 +85,11 @@ do elif [[ $para == --bind_core* ]]; then bind_core=`echo ${para#*=}` name_bind="_bindcore" + elif [[ $para == --conda_name* ]];then + conda_name=`echo ${para#*=}` + source set_conda.sh + echo "conda_name:$conda_name" + source activate $conda_name fi done