diff --git a/dataflow/gpu-examples/tensorflow-landsat-prime/Dockerfile b/dataflow/gpu-examples/tensorflow-landsat-prime/Dockerfile index 2cabbc313a..feb0e563d3 100644 --- a/dataflow/gpu-examples/tensorflow-landsat-prime/Dockerfile +++ b/dataflow/gpu-examples/tensorflow-landsat-prime/Dockerfile @@ -15,7 +15,7 @@ # NVIDIA CUDA container: https://catalog.ngc.nvidia.com/orgs/nvidia/containers/cuda # Supported NVIDIA images: https://gitlab.com/nvidia/container-images/cuda/blob/master/doc/supported-tags.md # TensorFlow/CUDA compatibility: https://www.tensorflow.org/install/source#gpu -FROM nvcr.io/nvidia/cuda:12.5.1-cudnn-runtime-ubuntu22.04 +FROM nvcr.io/nvidia/cuda:13.3.0-cudnn-runtime-ubuntu22.04 WORKDIR /pipeline diff --git a/dataflow/gpu-examples/tensorflow-landsat/Dockerfile b/dataflow/gpu-examples/tensorflow-landsat/Dockerfile index a0a685366b..0de95ac9c9 100644 --- a/dataflow/gpu-examples/tensorflow-landsat/Dockerfile +++ b/dataflow/gpu-examples/tensorflow-landsat/Dockerfile @@ -15,7 +15,7 @@ # NVIDIA CUDA container: https://catalog.ngc.nvidia.com/orgs/nvidia/containers/cuda # Supported NVIDIA images: https://gitlab.com/nvidia/container-images/cuda/blob/master/doc/supported-tags.md # TensorFlow/CUDA compatibility: https://www.tensorflow.org/install/source#gpu -FROM nvcr.io/nvidia/cuda:12.5.1-cudnn-runtime-ubuntu22.04 +FROM nvcr.io/nvidia/cuda:13.3.0-cudnn-runtime-ubuntu22.04 WORKDIR /pipeline diff --git a/dataflow/gpu-examples/tensorflow-minimal/Dockerfile b/dataflow/gpu-examples/tensorflow-minimal/Dockerfile index a0a685366b..0de95ac9c9 100644 --- a/dataflow/gpu-examples/tensorflow-minimal/Dockerfile +++ b/dataflow/gpu-examples/tensorflow-minimal/Dockerfile @@ -15,7 +15,7 @@ # NVIDIA CUDA container: https://catalog.ngc.nvidia.com/orgs/nvidia/containers/cuda # Supported NVIDIA images: https://gitlab.com/nvidia/container-images/cuda/blob/master/doc/supported-tags.md # TensorFlow/CUDA compatibility: https://www.tensorflow.org/install/source#gpu -FROM nvcr.io/nvidia/cuda:12.5.1-cudnn-runtime-ubuntu22.04 +FROM nvcr.io/nvidia/cuda:13.3.0-cudnn-runtime-ubuntu22.04 WORKDIR /pipeline