Merge pull request #145 from didi/zh794390558-patch-1

Update install-delta.sh
Delta-ML · Oct 23, 2019 · 1872e50 · 1872e50
2 parents 15045e7 + 84bf4ae
commit 1872e50
Showing 1 changed file with 4 additions and 4 deletions.
diff --git a/tools/install/install-delta.sh b/tools/install/install-delta.sh
@@ -26,9 +26,9 @@ else
 fi
 
 if [ $2 == 'gpu' ];then
-  TF='tensorflow-gpu=${TF_VER}'
+  TF="tensorflow-gpu=${TF_VER}"
 elif [ $2 == 'cpu' ];then
-  TF='tensorflow={TF_VER}'
+  TF="tensorflow=${TF_VER}"
 else
   echo ${USAGE}
   exit 1
@@ -45,9 +45,9 @@ CONDA_ENV=delta-py${PY_VER}-tf${TF_VER}
 if conda search ${TF} | grep "No match"; then
   echo "Conda: do not have ${TF} for current platform."
   if [ $2 == 'gpu' ];then
-    TF='tensorflow-gpu=${TF_VER}'
+    TF="tensorflow-gpu=${TF_VER}"
   else
-    TF='tensorflow=${TF_VER}'
+    TF="tensorflow=${TF_VER}"
   fi
   echo "Conda: install ${TF} instead."
   CONDA_ENV=delta-py${PY_VER}-tf${TF_VER}