diff --git a/.azure/gpu-pipeline.yml b/.azure/gpu-pipeline.yml
index 56ad6ffa333..a55652b3785 100644
--- a/.azure/gpu-pipeline.yml
+++ b/.azure/gpu-pipeline.yml
@@ -18,6 +18,15 @@ pr:
 
 jobs:
   - job: pytest
+    strategy:
+      matrix:
+        'PyTorch - LTS':
+          PyTorch: '1.8.1'
+          CUDA: '11.1'
+        'PyTorch - stable':
+          PyTorch: '1.11.0'
+          CUDA: '11.3'
+      #maxParallel: '2'
     # how long to run the job before automatically cancelling
     timeoutInMinutes: "55"
     # how much time to give 'run always even if cancelled tasks' before stopping them
@@ -26,7 +35,7 @@ jobs:
     pool: azure-gpus-spot
 
     container:
-      image: "pytorch/pytorch:1.8.1-cuda11.1-cudnn8-runtime"
+      image: "pytorch/pytorch:$(PyTorch)-cuda$(CUDA)-cudnn8-runtime"
       options: "--runtime=nvidia -e NVIDIA_VISIBLE_DEVICES=all --name ci-container -v /usr/bin/docker:/tmp/docker:ro"
 
     workspace:
diff --git a/.github/workflows/ci_schema.yml b/.github/workflows/ci_schema.yml
index 1e1f6a6a7b4..a0d08fc75dc 100644
--- a/.github/workflows/ci_schema.yml
+++ b/.github/workflows/ci_schema.yml
@@ -22,5 +22,6 @@ jobs:
           check-jsonschema .github/actions/*/*.yml --builtin-schema "github-actions"
 
       - name: Azure Pipelines
-        run: |
-          check-jsonschema .azure/*.yml --schemafile "https://raw.githubusercontent.com/microsoft/azure-pipelines-vscode/v1.188.1/service-schema.json"
+        env:
+          SCHEMA_FILE: https://raw.githubusercontent.com/microsoft/azure-pipelines-vscode/v1.204.0/service-schema.json
+        run: check-jsonschema .azure/*.yml --schemafile "$SCHEMA_FILE"