Skip to content

Commit cd9f135

Browse files
committed
Force upgrade.
1 parent 6bf2ae0 commit cd9f135

File tree

1 file changed

+1
-0
lines changed

1 file changed

+1
-0
lines changed

.azure/gpu-tests.yml

Lines changed: 1 addition & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -73,6 +73,7 @@ jobs:
7373
CUDA_VERSION_MM=$(python -c "import torch ; print(''.join(map(str, torch.version.cuda.split('.')[:2])))")
7474
pip install "bagua-cuda$CUDA_VERSION_MM>=0.9.0"
7575
pip install -e .[strategies]
76+
pip install deepspeed==0.6.4 # TODO: remove when docker images are upgraded
7677
pip install --requirement requirements/pytorch/devel.txt
7778
pip list
7879
env:

0 commit comments

Comments
 (0)