Replace tensorflow-gpu by tensorflow in Travis build
@@ -3,6 +3,9 @@ python:
- "3.5"
before_install:
- pip install -e .[tests]
+ # Replace tensorflow-gpu by tensorflow, see https://github.com/tensorflow/addons/issues/532
+ - pip install tensorflow="$(pip show tensorflow-gpu | grep Version | cut -f2 -d' ')"
+ - pip uninstall -y tensorflow-gpu
before_script:
- wget https://s3.amazonaws.com/opennmt-models/transliteration-aren-v2.tar.gz
- mkdir -p $TRAVIS_BUILD_DIR/testdata
