Skip to content

Commit 7bec0f1

Browse files
committed
add service information for nni, mlflow
1 parent 1b49379 commit 7bec0f1

File tree

1 file changed

+2
-2
lines changed

1 file changed

+2
-2
lines changed

vendor/ngc-pytorch/Dockerfile.20.12-py3

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

209210
RUN jupyter nbextensions_configurator enable && \
@@ -246,8 +247,7 @@ LABEL ai.backend.kernelspec="1" \
246247
ai.backend.base-distro="ubuntu16.04" \
247248
ai.backend.runtime-type="python" \
248249
ai.backend.runtime-path="/opt/conda/bin/python3" \
249-
ai.backend.service-ports="ipython:pty:3000,jupyter:http:8080,jupyterlab:http:8090,vscode:http:8180,tensorboard:http:6006"
250-
250+
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"
251251

252252
WORKDIR /home/work
253253
# vim: ft=dockerfile

0 commit comments

Comments
 (0)