Update tensorflow (#3418)

Azure · Oct 15, 2024 · 74539b1 · 74539b1
1 parent 32f360a
commit 74539b1
Showing 1 changed file with 1 addition and 1 deletion.
diff --git 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