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",
     ")"
    ]
   },