chore(deps): update nvcr.io/nvidia/cuda docker tag to v11.4.1 (Google…

…CloudPlatform#6516) Co-authored-by: Charles Engelke <[email protected]>
luigibertaco · Aug 5, 2021 · 9c71acb · 9c71acb
1 parent dbe515c
commit 9c71acb
Show file tree

Hide file tree

Showing 2 changed files with 2 additions and 2 deletions.
diff --git a/dataflow/gpu-examples/tensorflow-landsat/Dockerfile b/dataflow/gpu-examples/tensorflow-landsat/Dockerfile
@@ -16,7 +16,7 @@
 #   https://www.tensorflow.org/install/source#gpu
 # For a list of all the nvidia images:
 #   https://ngc.nvidia.com/catalog/containers/nvidia:cuda/tags
-FROM nvcr.io/nvidia/cuda:11.3.1-cudnn8-runtime-ubuntu20.04
+FROM nvcr.io/nvidia/cuda:11.4.1-cudnn8-runtime-ubuntu20.04
 
 WORKDIR /pipeline
 

diff --git a/dataflow/gpu-examples/tensorflow-minimal/Dockerfile b/dataflow/gpu-examples/tensorflow-minimal/Dockerfile
@@ -16,7 +16,7 @@
 #   https://www.tensorflow.org/install/source#gpu
 # For a list of all the nvidia images:
 #   https://ngc.nvidia.com/catalog/containers/nvidia:cuda/tags
-FROM nvcr.io/nvidia/cuda:11.3.1-cudnn8-runtime-ubuntu20.04
+FROM nvcr.io/nvidia/cuda:11.4.1-cudnn8-runtime-ubuntu20.04
 
 WORKDIR /pipeline