We read every piece of feedback, and take your input very seriously.
To see all available qualifiers, see our documentation.
There was an error while loading. Please reload this page.
1 parent 19bd62b commit 48a4e89Copy full SHA for 48a4e89
python-ff/Dockerfile.20.08-py36-cuda10.1
@@ -8,7 +8,7 @@ ENV TORCHVISION_VERSION=0.6.1
8
ENV TORCHAUDIO_VERSION=0.5.1
9
ENV TORCHTEXT_VERSION=0.6.0
10
ENV TENSORBOARDX_VERSION=2.1
11
-ENV MXNET_VERSION=1.6.0
+ENV MXNET_VERSION=1.6.0.post0
12
ARG CUDA=10.1
13
ARG LIBNVINFER=6.0.1-1
14
ARG LIBNVINFER_MAJOR_VERSION=6
@@ -99,6 +99,9 @@ RUN python3 -m pip install --no-cache-dir --upgrade \
99
nni==1.7 \
100
scikit-nni==0.2.1
101
102
+# Install MLFlow
103
+RUN python3 -m pip install --no-cache-dir mlflow
104
+
105
# Install ipython kernelspec
106
Run python3 -m ipykernel install --display-name "Lablup FF 20.08 on Python 3.6 (CUDA 10.1)" && \
107
cat /usr/local/share/jupyter/kernels/python3/kernel.json
0 commit comments