Skip to content

Commit d174b15

Browse files
committed
update horovod for mxnet
1 parent 9d26f34 commit d174b15

File tree

1 file changed

+1
-5
lines changed

1 file changed

+1
-5
lines changed

commons/Dockerfile.base.py38-cuda11.2

Lines changed: 1 addition & 5 deletions
Original file line numberDiff line numberDiff line change
@@ -427,7 +427,7 @@ RUN python3 -m pip install --extra-index-url \
427427
# Install Horovod, temporarily using CUDA stubs
428428
RUN ldconfig /usr/local/cuda/targets/x86_64-linux/lib/stubs && \
429429
HOROVOD_GPU_ALLREDUCE=NCCL HOROVOD_GPU_BROADCAST=NCCL HOROVOD_NCCL_LINK=SHARED \
430-
HOROVOD_WITH_TENSORFLOW=1 HOROVOD_WITH_PYTORCH=1 HOROVOD_WITHOUT_MXNET=1\
430+
HOROVOD_WITH_TENSORFLOW=1 HOROVOD_WITH_PYTORCH=1 HOROVOD_WITH_MXNET=1\
431431
HOROVOD_GPU=CUDA \
432432
python3 -m pip install --no-cache-dir horovod==0.23.0 && \
433433
ldconfig
@@ -456,10 +456,6 @@ COPY policy.yml /etc/backend.ai/jail/policy.yml
456456
COPY ./service-defs /etc/backend.ai/service-defs
457457
COPY ./runner-scripts/bootstrap.sh runner-scripts/setup_multinode.py /opt/container/
458458

459-
# Install ipython kernelspec
460-
RUN python3 -m ipykernel install --display-name "TensorFlow 2.7 on Python 3.8 (CUDA 11.2)" && \
461-
cat /usr/local/share/jupyter/kernels/python3/kernel.json
462-
463459
# Clean up
464460
RUN apt autoclean && \
465461
ln -s /usr/local/cuda-11.2/targets/x86_64-linux/lib/libcusolver.so.11 /usr/local/cuda-11.2/targets/x86_64-linux/lib/libcusolver.so.10 && \

0 commit comments

Comments
 (0)