update tensorflow variable initialization to new api

rasbt · Jan 23, 2017 · 391e72d · 391e72d
1 parent d3145f6
commit 391e72d
Showing 1 changed file with 3 additions and 3 deletions.
diff --git a/ci/.travis_install.sh b/ci/.travis_install.sh
@@ -35,9 +35,9 @@ python -c "import numpy; print('numpy %s' % numpy.__version__)"
 python -c "import scipy; print('scipy %s' % scipy.__version__)"
 
 if [ "${TENSORFLOW}" = "true" ]; then
-    pip uninstall protobuf
-    conda uninstall protobuf
-    conda install -c conda-forge tensorflow
+    #conda install -c conda-forge tensorflow
+    export TF_BINARY_URL=https://storage.googleapis.com/tensorflow/linux/gpu/tensorflow_gpu-0.12.1-cp35-cp35m-linux_x86_64.whl
+    sudo pip install --ignore-installed --upgrade $TF_BINARY_URL
     python -c "import tensorflow; print('tensorflow %s' % tensorflow.__version__)";
 else
     python -c "import sklearn; print('sklearn %s' % sklearn.__version__)";