From 1eca0e84d1a361045febb128069c307d8cc0217a Mon Sep 17 00:00:00 2001
From: Jeff Shepherd <jeffshep@microsoft.com>
Date: Sat, 12 Oct 2024 16:12:41 -0700
Subject: [PATCH] Update tensorflow

---
 .../single-step/tensorflow/mnist-distributed-horovod/job.yml    | 2 +-
 1 file changed, 1 insertion(+), 1 deletion(-)

diff --git a/cli/jobs/single-step/tensorflow/mnist-distributed-horovod/job.yml b/cli/jobs/single-step/tensorflow/mnist-distributed-horovod/job.yml
index 856c9cfe160..4ce38729cea 100644
--- a/cli/jobs/single-step/tensorflow/mnist-distributed-horovod/job.yml
+++ b/cli/jobs/single-step/tensorflow/mnist-distributed-horovod/job.yml
@@ -5,7 +5,7 @@ command: >-
   --epochs ${{inputs.epochs}}
 inputs:
   epochs: 1
-environment: azureml:AzureML-tensorflow-2.12-cuda11@latest
+environment: azureml:AzureML-tensorflow-2.16-cuda11@latest
 compute: azureml:gpu-cluster
 resources:
   instance_count: 2