Change the cloud build to use CUDA 9-0 instead of 9-1 (#2008)

googledatalab · May 10, 2018 · f36bbd0 · f36bbd0
1 parent bdffef9
commit f36bbd0
Showing 1 changed file with 2 additions and 2 deletions.
diff --git 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'