From 7949bb9b7f97935f59c1717be62b896e18f68699 Mon Sep 17 00:00:00 2001 From: Alexey Volkov Date: Mon, 16 Dec 2019 14:25:38 -0800 Subject: [PATCH] Build - Fix building TF images (#2736) --- .cloudbuild.yaml | 7 ++++++- components/release.sh | 2 +- 2 files changed, 7 insertions(+), 2 deletions(-) diff --git a/.cloudbuild.yaml b/.cloudbuild.yaml index d8c770f4729..cdd098b9366 100644 --- a/.cloudbuild.yaml +++ b/.cloudbuild.yaml @@ -100,6 +100,11 @@ steps: args: ['-c', 'cd /workspace/components/kubeflow/launcher && ./build_image.sh -p $PROJECT_ID -t $COMMIT_SHA'] id: 'buildLauncher' waitFor: ["-"] +- id: 'buildCpuTrainer' + name: 'gcr.io/cloud-builders/docker' + entrypoint: '/bin/bash' + args: ['-c', 'cd /workspace/components/kubeflow/dnntrainer && ./build_image.sh -p $PROJECT_ID -t $COMMIT_SHA -l ml-pipeline-kubeflow-tf-trainer -b 1.6.0'] + waitFor: ["-"] - name: 'gcr.io/cloud-builders/docker' entrypoint: '/bin/bash' args: ['-c', 'cd /workspace/components/kubeflow/dnntrainer && ./build_image.sh -p $PROJECT_ID -t $COMMIT_SHA -l ml-pipeline-kubeflow-tf-trainer-gpu -b 1.6.0-gpu'] @@ -177,7 +182,7 @@ images: - 'gcr.io/$PROJECT_ID/ml-pipeline-kubeflow-deployer:$COMMIT_SHA' - 'gcr.io/$PROJECT_ID/ml-pipeline-kubeflow-tf-trainer:$COMMIT_SHA' - 'gcr.io/$PROJECT_ID/ml-pipeline-kubeflow-tf-trainer-gpu:$COMMIT_SHA' -- 'gcr.io/$PROJECT_ID/ml-pipeline-kubeflow-tf:$COMMIT_SHA' +- 'gcr.io/$PROJECT_ID/ml-pipeline-kubeflow-tfjob:$COMMIT_SHA' # Images for the Dataproc-based pipeline components - 'gcr.io/$PROJECT_ID/ml-pipeline-dataproc-analyze:$COMMIT_SHA' diff --git a/components/release.sh b/components/release.sh index 88d768689fb..58e8b9f5fdf 100755 --- a/components/release.sh +++ b/components/release.sh @@ -27,7 +27,7 @@ images=( "ml-pipeline-kubeflow-deployer" "ml-pipeline-kubeflow-tf-trainer" "ml-pipeline-kubeflow-tf-trainer-gpu" - "ml-pipeline-kubeflow-tf" + "ml-pipeline-kubeflow-tfjob" "ml-pipeline-dataproc-analyze" "ml-pipeline-dataproc-create-cluster" "ml-pipeline-dataproc-delete-cluster"