Merge pull request #2331 from ROCmSoftwarePlatform/2.13-rt

[r2.13-rocm-enhanced] Dont install dev requirements in runtime mode
ROCm · Dec 19, 2023 · 3ed4f06 · 3ed4f06
2 parents 10c36c1 + 5715900
commit 3ed4f06
Showing 1 changed file with 7 additions and 2 deletions.
diff --git a/tensorflow/tools/tf_sig_build_dockerfiles/setup.python.sh b/tensorflow/tools/tf_sig_build_dockerfiles/setup.python.sh
@@ -99,5 +99,10 @@ curl https://bootstrap.pypa.io/get-pip.py -o get-pip.py
 python3 get-pip.py
 python3 -m pip install --no-cache-dir --upgrade pip
 
-# Disable the cache dir to save image space, and install packages
-python3 -m pip install --no-cache-dir -r $REQUIREMENTS -U
+if [[ $3 ]]; then
+    echo "Runtime mode"
+else
+    echo "Install Requirements"
+    # Disable the cache dir to save image space, and install packages
+    python3 -m pip install --no-cache-dir -r $REQUIREMENTS -U
+fi