-
Notifications
You must be signed in to change notification settings - Fork 26
/
.travis.yml
105 lines (94 loc) · 5.12 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
96
97
98
99
100
101
102
103
104
# After changing this file, check it on:
# http://lint.travis-ci.org/
os: linux
dist: xenial
sudo: false
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
osx_image: xcode9.2
language: generic
python: 3.6
env: WANTED_PYTHON_VERSION=3.6 TF_PREBUILT_VERSION=1.5.0
- os: osx
osx_image: xcode9.2
language: generic
python: 3.6
env: WANTED_PYTHON_VERSION=3.6 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
export SUDOH="sudo"
if [ "$WANTED_PYTHON_VERSION" == "2.7" ]; then
brew update # this swaps python versions and makes 3 the default one
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
elif [ "$WANTED_PYTHON_VERSION" == "3.6" ]; then
curl https://www.python.org/ftp/python/3.6.8/python-3.6.8-macosx10.9.pkg -s -o /tmp/python.pkg
${SUDOH} installer -pkg /tmp/python.pkg -target /
ls -l /Library/Frameworks/Python.framework/Versions/3.6/bin
export PATH=/Library/Frameworks/Python.framework/Versions/3.6/bin:$PATH
export PYTHON_EXE=python3
else
echo "Other MacOS Pythons are not supported at the moment"
fi
export CK_PLATFORM_NAME="generic-macos " # used later by CK
else
export WANTED_PYTHON_VERSION=$TRAVIS_PYTHON_VERSION # since Python is supported in Linux, get it from Travis
export PYTHON_EXE=python
export CK_PLATFORM_NAME="generic-linux " # used later by CK (note the trailing space to make the choice unique)
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
- rm -f `which pip` # the file is removable, but not writeable
- ${SUDOH} ${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 --tags=model,tf,mobilenet-v1,v1-1.0-224 # 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