diff --git a/sdk/python/endpoints/batch/deploy-models/huggingface-text-summarization/text-summarization-batch.ipynb b/sdk/python/endpoints/batch/deploy-models/huggingface-text-summarization/text-summarization-batch.ipynb
index 5046d5ddd1..3f1eaced44 100644
--- a/sdk/python/endpoints/batch/deploy-models/huggingface-text-summarization/text-summarization-batch.ipynb
+++ b/sdk/python/endpoints/batch/deploy-models/huggingface-text-summarization/text-summarization-batch.ipynb
@@ -612,7 +612,7 @@
    "source": [
     "environment = Environment(\n",
     "    conda_file=\"environment/torch200-conda.yaml\",\n",
-    "    image=\"mcr.microsoft.com/azureml/openmpi4.1.0-cuda11.8-cudnn8-ubuntu22.04:latest\",\n",
+    "    image=\"mcr.microsoft.com/azureml/openmpi5.0-cuda12.6-ubuntu24.04:latest\",\n",
     ")"
    ]
   },
diff --git a/sdk/python/endpoints/batch/deploy-models/openai-embeddings/deploy-and-test.ipynb b/sdk/python/endpoints/batch/deploy-models/openai-embeddings/deploy-and-test.ipynb
index 3885cd0e7c..6d1a705e3b 100644
--- a/sdk/python/endpoints/batch/deploy-models/openai-embeddings/deploy-and-test.ipynb
+++ b/sdk/python/endpoints/batch/deploy-models/openai-embeddings/deploy-and-test.ipynb
@@ -376,7 +376,7 @@
     "environment = Environment(\n",
     "    name=\"batch-openai-mlflow\",\n",
     "    conda_file=\"environment/conda.yaml\",\n",
-    "    image=\"mcr.microsoft.com/azureml/openmpi4.1.0-ubuntu20.04:latest\",\n",
+    "    image=\"mcr.microsoft.com/azureml/openmpi4.1.0-ubuntu22.04:latest\",\n",
     ")"
    ]
   },
@@ -796,7 +796,11 @@
    ]
   }
  ],
- "metadata": {},
+ "metadata": {
+  "language_info": {
+   "name": "python"
+  }
+ },
  "nbformat": 4,
  "nbformat_minor": 4
 }
diff --git a/sdk/python/endpoints/batch/deploy-pipelines/training-with-components/sdk-deploy-and-test.ipynb b/sdk/python/endpoints/batch/deploy-pipelines/training-with-components/sdk-deploy-and-test.ipynb
index bd990599b8..7cf518e63f 100644
--- a/sdk/python/endpoints/batch/deploy-pipelines/training-with-components/sdk-deploy-and-test.ipynb
+++ b/sdk/python/endpoints/batch/deploy-pipelines/training-with-components/sdk-deploy-and-test.ipynb
@@ -145,7 +145,7 @@
     "environment = Environment(\n",
     "    name=\"xgboost-sklearn-py38\",\n",
     "    description=\"An environment for models built with XGBoost and Scikit-learn.\",\n",
-    "    image=\"mcr.microsoft.com/azureml/openmpi4.1.0-ubuntu20.04:latest\",\n",
+    "    image=\"mcr.microsoft.com/azureml/openmpi4.1.0-ubuntu22.04:latest\",\n",
     "    conda_file=\"environment/conda.yml\",\n",
     ")"
    ]