From 74539b1830226a43e3aefa9d0bfa6c91c0bfd44d Mon Sep 17 00:00:00 2001
From: jeff-shepherd <39775772+jeff-shepherd@users.noreply.github.com>
Date: Tue, 15 Oct 2024 14:40:51 -0700
Subject: [PATCH] Update tensorflow (#3418)

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

diff --git a/cli/jobs/single-step/tensorflow/mnist/job.yml b/cli/jobs/single-step/tensorflow/mnist/job.yml
index fa3df23574a..76726421acf 100644
--- a/cli/jobs/single-step/tensorflow/mnist/job.yml
+++ b/cli/jobs/single-step/tensorflow/mnist/job.yml
@@ -2,7 +2,7 @@ $schema: https://azuremlschemas.azureedge.net/latest/commandJob.schema.json
 code: src
 command: >-
   python train.py
-environment: azureml:AzureML-tensorflow-2.12-cuda11@latest
+environment: azureml:AzureML-tensorflow-2.16-cuda11@latest
 compute: azureml:gpu-cluster
 display_name: tensorflow-mnist-example
 experiment_name: tensorflow-mnist-example