Skip to content

Commit 48a4e89

Browse files
committed
update: python-ff 20.08 image
1 parent 19bd62b commit 48a4e89

File tree

1 file changed

+4
-1
lines changed

1 file changed

+4
-1
lines changed

python-ff/Dockerfile.20.08-py36-cuda10.1

Lines changed: 4 additions & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -8,7 +8,7 @@ ENV TORCHVISION_VERSION=0.6.1
88
ENV TORCHAUDIO_VERSION=0.5.1
99
ENV TORCHTEXT_VERSION=0.6.0
1010
ENV TENSORBOARDX_VERSION=2.1
11-
ENV MXNET_VERSION=1.6.0
11+
ENV MXNET_VERSION=1.6.0.post0
1212
ARG CUDA=10.1
1313
ARG LIBNVINFER=6.0.1-1
1414
ARG LIBNVINFER_MAJOR_VERSION=6
@@ -99,6 +99,9 @@ RUN python3 -m pip install --no-cache-dir --upgrade \
9999
nni==1.7 \
100100
scikit-nni==0.2.1
101101

102+
# Install MLFlow
103+
RUN python3 -m pip install --no-cache-dir mlflow
104+
102105
# Install ipython kernelspec
103106
Run python3 -m ipykernel install --display-name "Lablup FF 20.08 on Python 3.6 (CUDA 10.1)" && \
104107
cat /usr/local/share/jupyter/kernels/python3/kernel.json

0 commit comments

Comments
 (0)