Skip to content

Commit 1b49379

Browse files
committed
Merge branch '20.12' of https://github.com/lablup/backend.ai-kernels into 20.12
2 parents 7a02e7e + 33f157d commit 1b49379

File tree

2 files changed

+7
-5
lines changed

2 files changed

+7
-5
lines changed

commons/Dockerfile.base.cuda11.1

Lines changed: 4 additions & 3 deletions
Original file line numberDiff line numberDiff line change
@@ -90,6 +90,7 @@ RUN rm -fr /var/lib/apt/lists/* && \
9090
libboost-filesystem-dev \
9191
libopenblas-dev liblapack-dev \
9292
xvfb \
93+
xz-utils \
9394
fonts-nanum \
9495
fonts-nanum-coding \
9596
fonts-nanum-extra \
@@ -100,9 +101,9 @@ RUN rm -fr /var/lib/apt/lists/* && \
100101
echo "/opt/intel/mkl/lib/intel64" >> /etc/ls.so.conf && \
101102
find /usr/local/cuda-${CUDA}/lib64/ -type f -name 'lib*_static.a' -not -name 'libcudart_static.a' -delete
102103

103-
RUN apt update && apt install curl xz-utils -y --no-install-recommends && NCCL_DOWNLOAD_SUM=34000cbe6a0118bfd4ad898ebc5f59bf5d532bbf2453793891fa3f1621e25653 && \
104-
curl -fsSL https://developer.nvidia.com/compute/machine-learning/nccl/secure/${NCCL}/ubuntu1804/x86_64/nccl-repo-ubuntu1804-${NCCL}-ga-cuda11.1_1-1_amd64.deb https://developer.nvidia.com/compute/machine-learning/nccl/secure/${NCCL}/agnostic/x64/nccl_${NCCL}-1+cuda11.1_x86_64.txz -O && \
105-
echo "$NCCL_DOWNLOAD_SUM nccl_2.8.3-1+cuda11.1_x86_64.txz" | sha256sum -c - && \
104+
# NCCL_DOWNLOAD_SUM=34000cbe6a0118bfd4ad898ebc5f59bf5d532bbf2453793891fa3f1621e25653
105+
# echo "$NCCL_DOWNLOAD_SUM nccl_2.8.3-1+cuda11.1_x86_64.txz" | sha256sum -c -
106+
RUN curl -fsSL https://developer.download.nvidia.com/compute/redist/nccl/v2.8/nccl_2.8.3-1+cuda11.1_x86_64.txz -O && \
106107
tar --no-same-owner --keep-old-files --lzma -xvf nccl_${NCCL}-1+cuda11.1_x86_64.txz -C /usr/local/cuda/lib64/ --strip-components=2 --wildcards '*/lib/libnccl.so.*' && \
107108
tar --no-same-owner --keep-old-files --lzma -xvf nccl_${NCCL}-1+cuda11.1_x86_64.txz -C /usr/lib/pkgconfig/ --strip-components=3 --wildcards '*/lib/pkgconfig/*' && \
108109
rm nccl_${NCCL}-1+cuda11.1_x86_64.txz && \

vendor/ngc-pytorch/Dockerfile.20.11-py3

Lines changed: 3 additions & 2 deletions
Original file line numberDiff line numberDiff line change
@@ -203,6 +203,7 @@ RUN ldconfig /usr/local/cuda/targets/x86_64-linux/lib/stubs && \
203203
RUN python3 -m pip install --no-cache-dir \
204204
mpi4py==3.0.3 \
205205
nni==1.9 \
206+
mlflow==1.13 \
206207
scikit-nni==0.2.1
207208

208209
RUN jupyter nbextensions_configurator enable && \
@@ -229,7 +230,7 @@ RUN apt autoclean && \
229230

230231
RUN /opt/conda/bin/python3 -m ipykernel install \
231232
--prefix=/opt/conda/ \
232-
--display-name "PyTorch 1.8 (NGC 20.11/Python 3.6 Conda) on Backend.AI" && \
233+
--display-name "PyTorch 1.7.1 (NGC 20.11/Python 3.6 Conda) on Backend.AI" && \
233234
cat /opt/conda/share/jupyter/kernels/python3/kernel.json
234235

235236
# Backend.AI specifics
@@ -245,7 +246,7 @@ LABEL ai.backend.kernelspec="1" \
245246
ai.backend.base-distro="ubuntu16.04" \
246247
ai.backend.runtime-type="python" \
247248
ai.backend.runtime-path="/opt/conda/bin/python3" \
248-
ai.backend.service-ports="ipython:pty:3000,jupyter:http:8080,jupyterlab:http:8090,vscode:http:8180,tensorboard:http:6006"
249+
ai.backend.service-ports="ipython:pty:3000,jupyter:http:8091,jupyterlab:http:8090,vscode:http:8180,tensorboard:http:6006,mlflow-ui:http:5000,nniboard:http:8080"
249250

250251

251252
WORKDIR /home/work

0 commit comments

Comments
 (0)