From 7c5d83c7e58a166d6425ba2b1b4a4931e2cf1c55 Mon Sep 17 00:00:00 2001
From: 15694147069 <591861959@qq.com>
Date: Thu, 3 Aug 2023 19:12:35 +0800
Subject: [PATCH] =?UTF-8?q?=E6=B7=BB=E5=8A=A0conda=E9=80=82=E9=85=8D?=
MIME-Version: 1.0
Content-Type: text/plain; charset=UTF-8
Content-Transfer-Encoding: 8bit

---
 .../ResNet101_ID0063_for_TensorFlow/test/set_conda.sh        | 2 ++
 .../test/train_performance_1p.sh                             | 5 +++++
 .../test/train_performance_8p.sh                             | 5 +++++
 3 files changed, 12 insertions(+)
 create mode 100644 TensorFlow/built-in/cv/image_classification/ResNet101_ID0063_for_TensorFlow/test/set_conda.sh

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 000000000..febb0fa34
--- /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 259868fe1..fe5372a9b 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 3037cc1b6..cbcbc37e7 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
 
-- 
Gitee