From f7d3505e555996fc21aeb9edd7567c61029e637d Mon Sep 17 00:00:00 2001 From: Michael Oliver Date: Thu, 15 Jun 2017 12:27:26 -0700 Subject: [PATCH] Add CNTK to travis build --- .travis.yml | 20 ++++++++++++++++++++ 1 file changed, 20 insertions(+) diff --git a/.travis.yml b/.travis.yml index 3edc3ee..69d2d0c 100644 --- a/.travis.yml +++ b/.travis.yml @@ -13,6 +13,10 @@ matrix: env: KERAS_BACKEND=theano - python: 3.5 env: KERAS_BACKEND=theano + - python: 2.7 + env: KERAS_BACKEND=cntk + - python: 3.5 + env: KERAS_BACKEND=cntk install: # code below is taken from http://conda.pydata.org/docs/travis.html # We do this conditionally because it saves us some downloading if the @@ -49,6 +53,22 @@ install: # install TensorFlow (CPU) - pip install tensorflow + # install cntk + - if [[ "$TRAVIS_PYTHON_VERSION" == "2.7" ]]; then + pip install https://cntk.ai/PythonWheel/CPU-Only/cntk-2.0-cp27-cp27mu-linux_x86_64.whl; + elif [[ "$TRAVIS_PYTHON_VERSION" == "3.5" ]]; then + pip install https://cntk.ai/PythonWheel/CPU-Only/cntk-2.0-cp35-cp35m-linux_x86_64.whl; + fi + + #install open mpi + - rm -rf ~/mpi + - mkdir ~/mpi + - pushd ~/mpi + - wget http://cntk.ai/PythonWheel/ForKeras/depends/openmpi_1.10-3.zip + - unzip ./openmpi_1.10-3.zip + - sudo dpkg -i openmpi_1.10-3.deb + - popd + # command to run tests script: # run keras backend init to initialize backend config