diff --git a/azure/azure-pr-pipeline.yml b/azure/azure-pr-pipeline.yml
index 5ef4aac..da882f4 100644
--- a/azure/azure-pr-pipeline.yml
+++ b/azure/azure-pr-pipeline.yml
@@ -40,4 +40,5 @@ extends:
           - template: templates/e2e.yml
             parameters:
               test_type: e2e
+              python_version: ${{ variables.python_version }}
 
diff --git a/azure/azure-release-pipeline.yml b/azure/azure-release-pipeline.yml
index 37937fc..fee731a 100644
--- a/azure/azure-release-pipeline.yml
+++ b/azure/azure-release-pipeline.yml
@@ -40,18 +40,22 @@ extends:
           - template: templates/e2e.yml
             parameters:
               test_type: smoketest
+              python_version: ${{ variables.python_version }}
       - environment: internal-dev-sandbox
         post_deploy:
           - template: templates/e2e.yml
             parameters:
               test_type: smoketest
+              python_version: ${{ variables.python_version }}
       - environment: internal-qa
         post_deploy:
           - template: templates/e2e.yml
             parameters:
               test_type: e2e
+              python_version: ${{ variables.python_version }}
       - environment: internal-qa-sandbox
         post_deploy:
           - template: templates/e2e.yml
             parameters:
               test_type: smoketest
+              python_version: ${{ variables.python_version }}
diff --git a/azure/templates/e2e.yml b/azure/templates/e2e.yml
index fa30704..e60dc24 100644
--- a/azure/templates/e2e.yml
+++ b/azure/templates/e2e.yml
@@ -1,7 +1,15 @@
 parameters:
   - name: test_type
 
+  - name: python_version
+    type: string
+    default: ''
+
 steps:
+    - task: UsePythonVersion@0
+      displayName: "Use Python ${{ parameters.python_version }}"
+      inputs:
+        versionSpec: ${{ parameters.python_version }}
 
     - task: s3-cache-action@1
       inputs: