File tree Expand file tree Collapse file tree 1 file changed +1
-5
lines changed Expand file tree Collapse file tree 1 file changed +1
-5
lines changed Original file line number Diff line number Diff line change @@ -427,7 +427,7 @@ RUN python3 -m pip install --extra-index-url \
427
427
# Install Horovod, temporarily using CUDA stubs
428
428
RUN ldconfig /usr/local/cuda/targets/x86_64-linux/lib/stubs && \
429
429
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\
431
431
HOROVOD_GPU=CUDA \
432
432
python3 -m pip install --no-cache-dir horovod==0.23.0 && \
433
433
ldconfig
@@ -456,10 +456,6 @@ COPY policy.yml /etc/backend.ai/jail/policy.yml
456
456
COPY ./service-defs /etc/backend.ai/service-defs
457
457
COPY ./runner-scripts/bootstrap.sh runner-scripts/setup_multinode.py /opt/container/
458
458
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
-
463
459
# Clean up
464
460
RUN apt autoclean && \
465
461
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 && \
You can’t perform that action at this time.
0 commit comments