From 299e238805cedf508ab94bd090bd97459566e0cb Mon Sep 17 00:00:00 2001
From: SagarikaKengunte <56979207+SagarikaKengunte@users.noreply.github.com>
Date: Fri, 24 Jan 2025 15:44:48 -0800
Subject: [PATCH] Update imagenet-classifier-batch SDK notebook to use the
 newer tensorflow base image (#3498)

---
 .../imagenet-classifier/imagenet-classifier-batch.ipynb         | 2 +-
 1 file changed, 1 insertion(+), 1 deletion(-)

diff --git a/sdk/python/endpoints/batch/deploy-models/imagenet-classifier/imagenet-classifier-batch.ipynb b/sdk/python/endpoints/batch/deploy-models/imagenet-classifier/imagenet-classifier-batch.ipynb
index 49b5ac40525..adece0ca1d0 100644
--- a/sdk/python/endpoints/batch/deploy-models/imagenet-classifier/imagenet-classifier-batch.ipynb
+++ b/sdk/python/endpoints/batch/deploy-models/imagenet-classifier/imagenet-classifier-batch.ipynb
@@ -693,7 +693,7 @@
     "environment = Environment(\n",
     "    name=\"tensorflow212-cuda11-gpu\",\n",
     "    conda_file=\"environment/conda.yaml\",\n",
-    "    image=\"mcr.microsoft.com/azureml/curated/tensorflow-2.12-cuda11:latest\",\n",
+    "    image=\"mcr.microsoft.com/azureml/curated/tensorflow-2.16-cuda12:latest\",\n",
     ")"
    ]
   },