Fix old cuda image removed from docker (#201)

Philmod · web-flow · commit 6387cb5c8d73 · 2023-07-05T16:39:27.000-04:00
http://b/290035572
diff --git a/gpu.Dockerfile b/gpu.Dockerfile
@@ -1,5 +1,5 @@
 ARG BASE_TAG=staging
-FROM nvidia/cuda:11.7.0-cudnn8-devel-ubuntu18.04 AS nvidia
+FROM nvidia/cuda:11.7.1-cudnn8-devel-ubuntu18.04 AS nvidia
 FROM gcr.io/kaggle-images/rstats:${BASE_TAG}
 ARG ncpus=1
 
