Skip to content
New issue

Have a question about this project? Sign up for a free GitHub account to open an issue and contact its maintainers and the community.

By clicking “Sign up for GitHub”, you agree to our terms of service and privacy statement. We’ll occasionally send you account related emails.

Already on GitHub? Sign in to your account

stable, dev PyTorch in Dockerfile and conda gh actions #3074

Merged
merged 24 commits into from
Sep 17, 2020
Merged
Prev Previous commit
Next Next commit
use HOROVOD_GPU_OPERATIONS env variable
ydcjeff authored and Borda committed Sep 17, 2020
commit c2feb967274fcf7fa4cc88c32db5efa808b15085
3 changes: 1 addition & 2 deletions dockers/base-cuda/Dockerfile
Original file line number Diff line number Diff line change
@@ -17,8 +17,7 @@ ARG CONDA_VERSION=4.7.12

SHELL ["/bin/bash", "-c"]

ENV HOROVOD_GPU_ALLREDUCE=NCCL
ENV HOROVOD_GPU_BROADCAST=NCCL
ENV HOROVOD_GPU_OPERATIONS=NCCL
ENV HOROVOD_WITH_PYTORCH=1
ENV HOROVOD_WITHOUT_TENSORFLOW=1
ENV HOROVOD_WITHOUT_MXNET=1