diff --git a/.azure/gpu-benchmark.yml b/.azure/gpu-benchmark.yml
index bcd925e1d2..26b4a544e0 100644
--- a/.azure/gpu-benchmark.yml
+++ b/.azure/gpu-benchmark.yml
@@ -68,14 +68,7 @@ jobs:
         pip list
       displayName: 'Image info & NVIDIA'
 
-    - bash: |
-        python .actions/assistant.py requirements_prune_pkgs --packages [horovod,bagua,colossalai] --req_files [requirements/pytorch/strategies.txt]
-
-        PYTORCH_VERSION=$(python -c "import torch; print(torch.__version__.split('+')[0])")
-        python ./requirements/pytorch/adjust-versions.py requirements/pytorch/base.txt ${PYTORCH_VERSION}
-      displayName: 'Adjust dependencies'
-
-    - bash: pip install -e .[dev,strategies,examples] --find-links ${TORCH_URL}
+    - bash: pip install -e .[dev] --find-links ${TORCH_URL}
       env:
         PACKAGE_NAME: "pytorch"
         FREEZE_REQUIREMENTS: "1"