pip uninstall -y deepspeed

Author: Borda

.azure/gpu-tests-fabric.yml

@@ -104,6 +104,8 @@ jobs:
           pip install -U "lightning-utilities[cli]"
           # drop deepspeed since it is not supported by our minimal Torch requirements
           python -m lightning_utilities.cli requirements prune-pkgs --packages deepspeed --req_files requirements/fabric/strategies.txt
+          # uninstall deepspeed since some older docker images have it pre-installed
+          pip uninstall -y deepspeed
         condition: contains(variables['Agent.JobName'], 'oldest')
         displayName: "setting oldest dependencies"
 

.azure/gpu-tests-pytorch.yml

@@ -108,6 +108,8 @@ jobs:
           pip install -U "lightning-utilities[cli]"
           # drop deepspeed since it is not supported by our minimal Torch requirements
           python -m lightning_utilities.cli requirements prune-pkgs --packages deepspeed --req_files requirements/pytorch/strategies.txt
+          # uninstall deepspeed since some older docker images have it pre-installed
+          pip uninstall -y deepspeed
         condition: contains(variables['Agent.JobName'], 'oldest')
         displayName: "setting oldest dependencies"