forked from ctuning/ck-tensorflow
-
Notifications
You must be signed in to change notification settings - Fork 0
/
.travis.yml
96 lines (86 loc) · 4.63 KB
/
.travis.yml
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73
74
75
76
77
78
79
80
81
82
83
84
85
86
87
88
89
90
91
92
93
94
95
# After changing this file, check it on:
# http://lint.travis-ci.org/
os: linux
language: python
python:
- 2.7
- 3.6
env:
matrix:
- TF_PREBUILT_VERSION=1.4.0
- TF_PREBUILT_VERSION=1.5.0
- TF_PREBUILT_VERSION=1.7.0
- TF_PREBUILT_VERSION=1.8.0
matrix:
include:
- os: osx
language: generic
python: 2.7
env: WANTED_PYTHON_VERSION=2.7 TF_PREBUILT_VERSION=1.1.0
- os: osx
language: generic
python: 2.7
env: WANTED_PYTHON_VERSION=2.7 TF_PREBUILT_VERSION=1.4.0
- os: osx
language: generic
python: 2.7
env: WANTED_PYTHON_VERSION=2.7 TF_PREBUILT_VERSION=1.8.0
- os: osx
language: generic
python: 3.7
env: WANTED_PYTHON_VERSION=3.7 TF_PREBUILT_VERSION=1.4.0
- os: osx
language: generic
python: 3.7
env: WANTED_PYTHON_VERSION=3.7 TF_PREBUILT_VERSION=1.8.0
- os: linux
language: python
python: 2.7
env: TF_PREBUILT_VERSION=1.1.0
- os: linux
language: python
python: 3.5
env: TF_PREBUILT_VERSION=1.1.0
before_install:
- |
if [ $TRAVIS_OS_NAME == "osx" ]; then
brew update # this swaps python versions and makes 3 the default one
if [ "$WANTED_PYTHON_VERSION" == "2.7" ]; then
brew reinstall python\@2 || brew link --overwrite python\@2 # install and link python2 and pip2 to /usr/local/bin
export PATH=/usr/local/opt/python\@2/bin:$PATH
export PYTHON_EXE=python
else
brew reinstall python # install and link python3 and pip3 to /usr/local/bin
export PATH=/usr/local/opt/python/bin:$PATH
export PYTHON_EXE=python3
fi
export CK_PLATFORM_NAME="generic-macos " # used later by CK
else
sudo apt-get update -qq
sudo apt-get install python-pip
export WANTED_PYTHON_VERSION=$TRAVIS_PYTHON_VERSION # since Python is supported in Linux, get it from Travis
export CK_PLATFORM_NAME="generic-linux " # used later by CK (note the trailing space to make the choice unique)
export PYTHON_EXE=python
fi
install:
- CWD=`pwd`
- THIS_REPO_NAME=`basename $CWD`
- echo "TRAVIS_OS_NAME=${TRAVIS_OS_NAME}, WANTED_PYTHON_VERSION=${WANTED_PYTHON_VERSION}, TF_PREBUILT_VERSION=${TF_PREBUILT_VERSION}"
- echo "CWD=${CWD}, THIS_REPO_NAME=${THIS_REPO_NAME}, PATH=${PATH}"
- which ${PYTHON_EXE}
- ${PYTHON_EXE} --version
- ${PYTHON_EXE} -m pip install --ignore-installed --verbose pip setuptools # make sure pip is also up to date no matter what
- ${PYTHON_EXE} -m pip install -r requirements.txt # ck kernel itself and some essential components
- ck pull repo:${THIS_REPO_NAME} # ...also trigger pulling dependent repositories
- ck pull repo:ck-mlperf # a cyclic dependency that we don't want to create at this point
- echo "$CK_PLATFORM_NAME" | ck detect platform.os --update_platform_init # set the platform to generic-linux (1) or generic-macos (2)
- ck detect soft:compiler.python --full_path=`which ${PYTHON_EXE}` # pick the one Travis has set for us in the PATH
- ck install package:imagenet-2012-aux # one part of ImageNet
- ck install package:imagenet-2012-val-min-resized # another part of ImageNet
- ck install package:tensorflowmodel-mobilenet-v1-1.0-224-py # model for TF
- ck install package:lib-tensorflow-${TF_PREBUILT_VERSION}-cpu # prebuilt TF package
- ck install package --tags=lib,python-package,numpy # to prevent confusion with pre-installed version on Macs @ Travis
- ck install package --tags=lib,python-package,scipy --force_version=1.2.1 # to prevent confusion with pre-installed version on Macs @ Travis
- ck install package --tags=lib,python-package,pillow # to prevent confusion with pre-installed version on Macs @ Travis
script:
- ck run program:image-classification-tf-py --env.CK_IMAGE_FILE=$HOME/CK-TOOLS/dataset-imagenet-ilsvrc2012-val-min-resized/ILSVRC2012_val_00000060.JPEG