diff --git a/tools/ci_build/github/azure-pipelines/c-api-noopenmp-packaging-pipelines.yml b/tools/ci_build/github/azure-pipelines/c-api-noopenmp-packaging-pipelines.yml
index cae34d451c..4408e3d244 100644
--- a/tools/ci_build/github/azure-pipelines/c-api-noopenmp-packaging-pipelines.yml
+++ b/tools/ci_build/github/azure-pipelines/c-api-noopenmp-packaging-pipelines.yml
@@ -127,7 +127,7 @@ stages:
   jobs:
   - job: D1
     pool:
-      vmImage: ubuntu-latest
+      name: 'onnxruntime-Ubuntu2204-AMD-CPU'
     variables:
       MyVar: $[stageDependencies.Setup.Set_Variables.outputs['Set_Release_Version_Suffix.ReleaseVersionSuffix']]
       BuildDate: $[stageDependencies.Setup.Set_Variables.outputs['Set_Build_Date.BuildDate']]
@@ -149,7 +149,7 @@ stages:
   jobs:
   - job: Download_Java_Tools
     pool:
-      vmImage: ubuntu-latest
+      name: 'onnxruntime-Ubuntu2204-AMD-CPU'
     steps:
     - checkout: none
     - task: mspremier.PostBuildCleanup.PostBuildCleanup-task.PostBuildCleanup@3
diff --git a/tools/ci_build/github/azure-pipelines/py-package-test-pipeline.yml b/tools/ci_build/github/azure-pipelines/py-package-test-pipeline.yml
index acec6f501e..59c3427f88 100644
--- a/tools/ci_build/github/azure-pipelines/py-package-test-pipeline.yml
+++ b/tools/ci_build/github/azure-pipelines/py-package-test-pipeline.yml
@@ -50,7 +50,7 @@ stages:
         job_name: Test_LINUX_x86_64_Wheels
         itemPattern: '*/*manylinux*x86_64.whl'
         machine_pool:
-          vmImage: 'ubuntu-22.04'
+          name: 'onnxruntime-Ubuntu2204-AMD-CPU'
 
 # ****The following Stage depend on all previous tags. ***
 
diff --git a/tools/ci_build/github/azure-pipelines/stages/set_packaging_variables_stage.yml b/tools/ci_build/github/azure-pipelines/stages/set_packaging_variables_stage.yml
index 616dc5495d..3e2b3b585d 100644
--- a/tools/ci_build/github/azure-pipelines/stages/set_packaging_variables_stage.yml
+++ b/tools/ci_build/github/azure-pipelines/stages/set_packaging_variables_stage.yml
@@ -7,7 +7,7 @@ stages:
   jobs:
   - job: Set_Variables
     pool:
-      vmImage: ubuntu-latest
+      name: 'onnxruntime-Ubuntu2204-AMD-CPU'
     steps:
     - checkout: none
     - task: mspremier.PostBuildCleanup.PostBuildCleanup-task.PostBuildCleanup@3