We read every piece of feedback, and take your input very seriously.
To see all available qualifiers, see our documentation.
There was an error while loading. Please reload this page.
1 parent 6bf2ae0 commit cd9f135Copy full SHA for cd9f135
.azure/gpu-tests.yml
@@ -73,6 +73,7 @@ jobs:
73
CUDA_VERSION_MM=$(python -c "import torch ; print(''.join(map(str, torch.version.cuda.split('.')[:2])))")
74
pip install "bagua-cuda$CUDA_VERSION_MM>=0.9.0"
75
pip install -e .[strategies]
76
+ pip install deepspeed==0.6.4 # TODO: remove when docker images are upgraded
77
pip install --requirement requirements/pytorch/devel.txt
78
pip list
79
env:
0 commit comments