From f36bbd07e7aa2207fc26eeb9e7f32c9bd7382cbb Mon Sep 17 00:00:00 2001 From: Omar Jarjur Date: Thu, 10 May 2018 15:27:34 -0700 Subject: [PATCH] Change the cloud build to use CUDA 9-0 instead of 9-1 (#2008) --- tools/release/cloudbuild.yaml | 4 ++-- 1 file changed, 2 insertions(+), 2 deletions(-) diff --git a/tools/release/cloudbuild.yaml b/tools/release/cloudbuild.yaml index 1dc36419a..bead3405b 100644 --- a/tools/release/cloudbuild.yaml +++ b/tools/release/cloudbuild.yaml @@ -81,10 +81,10 @@ steps: ## Second, we build the GPU base image - name: 'gcr.io/cloud-builders/docker' - args: ['pull', 'nvidia/cuda:9.1-cudnn7-devel-ubuntu16.04'] + args: ['pull', 'nvidia/cuda:9.0-cudnn7-devel-ubuntu16.04'] id: 'pullNvidiaUbuntu' - name: 'gcr.io/cloud-builders/docker' - args: ['tag', 'nvidia/cuda:9.1-cudnn7-devel-ubuntu16.04', 'datalab-external-base-image'] + args: ['tag', 'nvidia/cuda:9.0-cudnn7-devel-ubuntu16.04', 'datalab-external-base-image'] id: 'tagNvidiaUbuntu' waitFor: ['buildBase', 'pullNvidiaUbuntu'] - name: 'gcr.io/cloud-builders/docker'