diff --git a/TensorFlow/built-in/cv/image_segmentation/2Dattentionunet_ID0120_for_TensorFlow/test/set_conda.sh b/TensorFlow/built-in/cv/image_segmentation/2Dattentionunet_ID0120_for_TensorFlow/test/set_conda.sh
new file mode 100644
index 0000000000000000000000000000000000000000..55087d8622f46e055bb105a2acd1fe7006d4af07
--- /dev/null
+++ b/TensorFlow/built-in/cv/image_segmentation/2Dattentionunet_ID0120_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_segmentation/2Dattentionunet_ID0120_for_TensorFlow/test/train_RT2_performance_1p.sh b/TensorFlow/built-in/cv/image_segmentation/2Dattentionunet_ID0120_for_TensorFlow/test/train_RT2_performance_1p.sh
index 2b733e78a08c3eaea7a3cf120fb9fa0202bc26a7..5d93fcbd17710bedc6c1f905aa7c4862c9851e33 100644
--- a/TensorFlow/built-in/cv/image_segmentation/2Dattentionunet_ID0120_for_TensorFlow/test/train_RT2_performance_1p.sh
+++ b/TensorFlow/built-in/cv/image_segmentation/2Dattentionunet_ID0120_for_TensorFlow/test/train_RT2_performance_1p.sh
@@ -84,6 +84,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/TensorFlow2/built-in/nlp/Word2vec_ID2350_for_TensorFlow2.X/test/set_conda.sh b/TensorFlow2/built-in/nlp/Word2vec_ID2350_for_TensorFlow2.X/test/set_conda.sh
new file mode 100644
index 0000000000000000000000000000000000000000..55087d8622f46e055bb105a2acd1fe7006d4af07
--- /dev/null
+++ b/TensorFlow2/built-in/nlp/Word2vec_ID2350_for_TensorFlow2.X/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/TensorFlow2/built-in/nlp/Word2vec_ID2350_for_TensorFlow2.X/test/train_RT2_performance_1p.sh b/TensorFlow2/built-in/nlp/Word2vec_ID2350_for_TensorFlow2.X/test/train_RT2_performance_1p.sh
index c1c57ab0adb255a3f040df8700803b086e54f89e..22295475c9c2fe4bd8f27b55cb3271e8f38b9519 100644
--- a/TensorFlow2/built-in/nlp/Word2vec_ID2350_for_TensorFlow2.X/test/train_RT2_performance_1p.sh
+++ b/TensorFlow2/built-in/nlp/Word2vec_ID2350_for_TensorFlow2.X/test/train_RT2_performance_1p.sh
@@ -76,6 +76,11 @@ do
         fusion_off_file=`echo ${para#*=}`
     elif [[ $para == --auto_tune* ]];then
         auto_tune=`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