@@ -142,22 +142,25 @@ jobs:
142142 # pip install -r requirements/pytorch/strategies.txt --find-links ${TORCH_URL}
143143 pip list
144144
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
161164
162165 - name : Cache datasets
163166 uses : actions/cache@v3
0 commit comments