From 391e72ded34bdcfe98eb8d5b32fdaf18bb79d04a Mon Sep 17 00:00:00 2001 From: rasbt Date: Mon, 23 Jan 2017 02:48:39 -0500 Subject: [PATCH] update tensorflow variable initialization to new api --- ci/.travis_install.sh | 6 +++--- 1 file changed, 3 insertions(+), 3 deletions(-) diff --git a/ci/.travis_install.sh b/ci/.travis_install.sh index 3b376c17d..de5c752f5 100755 --- 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__)";