diff --git a/dataflow/gpu-examples/tensorflow-landsat-prime/Dockerfile b/dataflow/gpu-examples/tensorflow-landsat-prime/Dockerfile index a506a8727a7..eef9ec20f05 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:11.8.0-cudnn8-runtime-ubuntu22.04 +FROM nvcr.io/nvidia/cuda:12.2.2-cudnn8-runtime-ubuntu22.04 WORKDIR /pipeline diff --git a/dataflow/gpu-examples/tensorflow-landsat/Dockerfile b/dataflow/gpu-examples/tensorflow-landsat/Dockerfile index 39a836fdb0b..dd37e5c0e4b 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:11.8.0-cudnn8-runtime-ubuntu22.04 +FROM nvcr.io/nvidia/cuda:12.2.2-cudnn8-runtime-ubuntu22.04 WORKDIR /pipeline diff --git a/dataflow/gpu-examples/tensorflow-minimal/Dockerfile b/dataflow/gpu-examples/tensorflow-minimal/Dockerfile index e5f79f6e4ad..bc95c68a49b 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:11.8.0-cudnn8-runtime-ubuntu22.04 +FROM nvcr.io/nvidia/cuda:12.2.2-cudnn8-runtime-ubuntu22.04 WORKDIR /pipeline