tf 0.12

rasbt · Jan 23, 2017 · 7f99d92 · 7f99d92
1 parent 928f17f
commit 7f99d92
Showing 1 changed file with 6 additions and 2 deletions.
diff --git a/ci/.travis_install.sh b/ci/.travis_install.sh
@@ -36,9 +36,13 @@ python -c "import scipy; print('scipy %s' % scipy.__version__)"
 
 if [ "${TENSORFLOW}" = "true" ]; then
     #conda install -c conda-forge tensorflow
-    export TF_BINARY_URL=https://storage.googleapis.com/tensorflow/linux/cpu/tensorflow-0.12.1-cp35-cp35m-linux_x86_64.whl
-    sudo pip install --ignore-installed --upgrade $TF_BINARY_URL
+    if [ "${TRAVIS_PYTHON_VERSION}" == "2.7" ]; then
+        pip install https://storage.googleapis.com/tensorflow/linux/cpu/tensorflow-0.12.1-cp27-none-linux_x86_64.whl;
+    else
+        pip install https://storage.googleapis.com/tensorflow/linux/cpu/tensorflow-0.12.1-cp35-cp35m-linux_x86_64.whl;
+    fi
     python -c "import tensorflow; print('tensorflow %s' % tensorflow.__version__)";
+
 else
     python -c "import sklearn; print('sklearn %s' % sklearn.__version__)";
 fi