diff --git a/docker/cloudbuild.yaml b/docker/cloudbuild.yaml index a018da36..43c387fe 100644 --- a/docker/cloudbuild.yaml +++ b/docker/cloudbuild.yaml @@ -2,5 +2,9 @@ steps: - id: 'build-cpu-image' name: gcr.io/cloud-builders/docker args: ['build', '-f', './docker/cpu.Dockerfile', '-t', 'gcr.io/kaggle-images/python-simulations', '.'] -images: ['gcr.io/kaggle-images/python-simulations'] +- id: 'build-gpu-image' + name: gcr.io/cloud-builders/docker + args: ['build', '-f', './docker/gpu.Dockerfile', '-t', 'gcr.io/kaggle-gpu-images/python-simulations', '.'] +waitFor: ['-'] +images: ['gcr.io/kaggle-images/python-simulations', 'gcr.io/kaggle-gpu-images/python-simulations'] timeout: 24h \ No newline at end of file