@@ -142,22 +142,25 @@ jobs:
142
142
# pip install -r requirements/pytorch/strategies.txt --find-links ${TORCH_URL}
143
143
pip list
144
144
145
- - name : Reinstall Horovod if necessary
146
- if : runner.os != 'windows'
147
- env :
148
- HOROVOD_BUILD_ARCH_FLAGS : " -mfma"
149
- HOROVOD_WITHOUT_MXNET : 1
150
- HOROVOD_WITHOUT_TENSORFLOW : 1
151
- run : |
152
- HOROVOD_BUILT=$(python -c "import horovod.torch; horovod.torch.nccl_built(); print('SUCCESS')" || true)
153
- if [[ $HOROVOD_BUILT != "SUCCESS" ]]; then
154
- pip uninstall -y horovod
155
- grep "horovod" requirements/pytorch/strategies.txt > requirements/pytorch/horovod.txt
156
- pip install --no-cache-dir -r requirements/pytorch/horovod.txt
157
- fi
158
- horovodrun --check-build
159
- python -c "import horovod.torch"
160
- pip list
145
+ # - name: Reinstall Horovod if necessary
146
+ # if: runner.os != 'windows'
147
+ # env:
148
+ # HOROVOD_BUILD_ARCH_FLAGS: "-mfma"
149
+ # HOROVOD_WITHOUT_MXNET: 1
150
+ # HOROVOD_WITHOUT_TENSORFLOW: 1
151
+ # run: |
152
+ # HOROVOD_BUILT=$(python -c "import horovod.torch; horovod.torch.nccl_built(); print('SUCCESS')" || true)
153
+ # if [[ $HOROVOD_BUILT != "SUCCESS" ]]; then
154
+ # pip uninstall -y horovod
155
+ # # grep "horovod" requirements/pytorch/strategies.txt > requirements/pytorch/horovod.txt
156
+ # # pip install --no-cache-dir -r requirements/pytorch/horovod.txt
157
+ #
158
+ # # we hardcode the version for testing due to difficulties installing and compiling the package
159
+ # pip install --no-cache-dir horovod==0.21.2
160
+ # fi
161
+ # horovodrun --check-build
162
+ # python -c "import horovod.torch"
163
+ # pip list
161
164
162
165
- name : Cache datasets
163
166
uses : actions/cache@v3
0 commit comments