Skip to content

Commit 7c9cfa8

Browse files
committed
fixed conflict library with Tensorboard 2.x
1 parent 27bc500 commit 7c9cfa8

File tree

1 file changed

+29
-22
lines changed

1 file changed

+29
-22
lines changed

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

Lines changed: 29 additions & 22 deletions
Original file line numberDiff line numberDiff line change
@@ -24,42 +24,50 @@ ENV PYTHONUNBUFFERED=1 \
2424
# Set default shell to /bin/bash
2525
SHELL ["/bin/bash", "-cu"]
2626

27-
RUN python3 -m pip install pip --no-cache-dir --upgrade \
27+
RUN python3 -m pip install --no-cache-dir --upgrade \
28+
gast==0.2.2 \
29+
'httplib2<0.18.0,>=0.8' \
30+
'coverage<5.0,>=4.5' \
31+
'mock<4.0,>=3.0' \
32+
tensorflow==${TENSORFLOW_VERSION} \
33+
tensorboard==2.1.1
34+
35+
RUN python3 -m pip install --no-cache-dir --upgrade \
36+
-q git+https://github.com/google-research/tensorflow_constrained_optimization
37+
RUN python3 -m pip install --no-cache-dir --upgrade \
38+
-q git+https://github.com/tensorflow/fairness-indicators
39+
RUN python3 -m pip install --no-cache-dir \
40+
cachetools==3.1.0 \
2841
tensorflow-transform==0.21.2 \
2942
tensorflow_model_analysis==0.21.6 \
3043
ngraph-tensorflow-bridge==0.18.0 \
31-
adanet==0.8.0
32-
33-
RUN python3 -m pip install pip --no-cache-dir --upgrade \
34-
pandas==1.0.4 \
35-
tensorflow-gpu==${TENSORFLOW_VERSION} \
36-
tensorboard==2.2.2 \
3744
tensorflow-metadata==0.21.2 \
3845
tensorflow-datasets==3.1.0 \
39-
tensorflow_probability==0.7.0 \
46+
tensorflow_probability==0.7.0 \
4047
tensorflow-addons==0.10.0 \
4148
neural-structured-learning==1.1.0 \
42-
tensorflow-model-optimization==0.3.0 \
49+
tensorflow-model-optimization==0.3.0
50+
RUN python3 -m pip install --no-cache-dir \
51+
tfx==0.21.4 \
52+
tfx-bsl==0.21.4 \
53+
tf-agents==0.5.0
54+
RUN python3 -m pip install --no-cache-dir \
55+
tensorflow_constrained_optimization \
4356
tensorflow-hub==0.8.0 \
4457
tensorflow_text==2.2.0 \
4558
tensorflow-graphics-gpu==1.0.0 \
4659
tensorflow-gan==2.0.0 \
47-
tensorflow-data-validation==0.21.4 \
48-
fairness-indicators==0.1.2 \
49-
tensorboard-plugin-fairness-indicators==0.0.5 \
50-
fairness==0.1.8 \
51-
tfx==0.21.4 \
52-
tfx-bsl==0.22.0 \
53-
tf-agents==0.5.0 \
54-
tensorflow-lattice-gpu==0.9.8 \
60+
tensorflow-data-validation==0.21.4
61+
RUN python3 -m pip install --no-cache-dir \
62+
jupyter-tensorboard==0.2.0 \
63+
tensorflow-plot==0.3.2 \
5564
tensorflow-lattice==2.0.4 \
5665
tensorflow_ranking==0.3.1 \
5766
tensorflow-compression==1.3 \
5867
tensor2tensor==1.15.6 \
5968
mesh-tensorflow==0.1.13 \
60-
tensorflow-io==0.13.0 \
61-
jupyter-tensorboard==0.2.0 \
6269
tensorwatch==0.9.1 \
70+
tensorflow-io==0.13.0 \
6371
jupyterlab-nvdashboard==0.3.1
6472

6573
RUN python3 -m pip install pip --no-cache-dir \
@@ -76,13 +84,12 @@ RUN python3 -m pip install --no-cache-dir mxnet-cu101==${MXNET_VERSION}
7684
RUN ldconfig /usr/local/cuda/targets/x86_64-linux/lib/stubs && \
7785
HOROVOD_GPU_ALLREDUCE=NCCL HOROVOD_GPU_BROADCAST=NCCL \
7886
HOROVOD_WITH_TENSORFLOW=1 HOROVOD_WITH_PYTORCH=1 HOROVOD_WITH_MXNET=1 \
79-
pip install --no-cache-dir horovod==0.19.4 && \
87+
pip install --no-cache-dir horovod==0.19.0 && \
8088
ldconfig
8189

8290
RUN python3 -m pip install --no-cache-dir --upgrade \
8391
mpi4py==3.0.3 \
84-
nni==1.6 \
85-
scikit-nni==0.2.1
92+
nni==1.6
8693

8794
# Install ipython kernelspec
8895
Run python3 -m ipykernel install --display-name "Lablup FF 20.06 on Python 3.6 (CUDA 10.1)" && \

0 commit comments

Comments
 (0)