forked from aws/deep-learning-containers
-
Notifications
You must be signed in to change notification settings - Fork 0
/
Copy pathbuildspec.yml
35 lines (33 loc) · 1008 Bytes
/
buildspec.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
version: 0.2
#TODO: Move all constatns to environment variables
phases:
install:
runtime-versions:
docker: 18
pre_build:
commands:
- start-dockerd
- echo Logging in to Amazon ECR...
- $(aws ecr get-login --no-include-email --region $AWS_DEFAULT_REGION)
- pip install -r src/requirements.txt
- bash src/setup.sh $FRAMEWORK
build:
commands:
- echo Build started on `date`
- echo Building the Docker image...
- |
if expr "${CODEBUILD_BUILD_ARN}" : ".*tensorflow-1" >/dev/null; then
python src/main.py --buildspec $FRAMEWORK/buildspec-tf1.yml --framework $FRAMEWORK
else
python src/main.py --buildspec $FRAMEWORK/buildspec.yml --framework $FRAMEWORK
fi
post_build:
commands:
- echo Build step finished on `date`
- echo Starting test jobs
- python src/start_testbuilds.py
artifacts:
files:
- $HOME/test_type_images.json
name: DLC_IMAGES_JSON
discard-paths: yes