From a6462014a708f04e556777f14405bef8e01d0e7d Mon Sep 17 00:00:00 2001 From: Bovard Doerschuk-Tiberi Date: Wed, 13 Dec 2023 10:39:01 -1000 Subject: [PATCH] build GPU image (#252) --- docker/cloudbuild.yaml | 6 +++++- 1 file changed, 5 insertions(+), 1 deletion(-) 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