diff --git a/TensorFlow/built-in/cv/image_classification/ResNet101_ID0063_for_TensorFlow/test/set_conda.sh b/TensorFlow/built-in/cv/image_classification/ResNet101_ID0063_for_TensorFlow/test/set_conda.sh new file mode 100644 index 0000000000000000000000000000000000000000..febb0fa34937160c9ad16c92dcb67cd4005e2f57 --- /dev/null +++ b/TensorFlow/built-in/cv/image_classification/ResNet101_ID0063_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/image_classification/ResNet101_ID0063_for_TensorFlow/test/train_performance_1p.sh b/TensorFlow/built-in/cv/image_classification/ResNet101_ID0063_for_TensorFlow/test/train_performance_1p.sh index 259868fe1d8cacc8be7d38d4ed7f31cdaad22821..fe5372a9bf9c0253de0ab61c50557a647035cecc 100644 --- a/TensorFlow/built-in/cv/image_classification/ResNet101_ID0063_for_TensorFlow/test/train_performance_1p.sh +++ b/TensorFlow/built-in/cv/image_classification/ResNet101_ID0063_for_TensorFlow/test/train_performance_1p.sh @@ -87,6 +87,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/image_classification/ResNet101_ID0063_for_TensorFlow/test/train_performance_8p.sh b/TensorFlow/built-in/cv/image_classification/ResNet101_ID0063_for_TensorFlow/test/train_performance_8p.sh index 3037cc1b6781a15ce987c557d4af3f82828f7364..cbcbc37e725f7f5e6b929a2651b4d1e159e44201 100644 --- a/TensorFlow/built-in/cv/image_classification/ResNet101_ID0063_for_TensorFlow/test/train_performance_8p.sh +++ b/TensorFlow/built-in/cv/image_classification/ResNet101_ID0063_for_TensorFlow/test/train_performance_8p.sh @@ -88,6 +88,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