Fix old cuda image removed from docker (#201)

http://b/290035572
Kaggle · Jul 5, 2023 · 6387cb5 · 6387cb5
1 parent 0441e26
commit 6387cb5
Showing 1 changed file with 1 addition and 1 deletion.
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