@@ -24,42 +24,50 @@ ENV PYTHONUNBUFFERED=1 \
24
24
# Set default shell to /bin/bash
25
25
SHELL ["/bin/bash" , "-cu" ]
26
26
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 \
28
41
tensorflow-transform==0.21.2 \
29
42
tensorflow_model_analysis==0.21.6 \
30
43
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 \
37
44
tensorflow-metadata==0.21.2 \
38
45
tensorflow-datasets==3.1.0 \
39
- tensorflow_probability==0.7.0 \
46
+ tensorflow_probability==0.7.0 \
40
47
tensorflow-addons==0.10.0 \
41
48
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 \
43
56
tensorflow-hub==0.8.0 \
44
57
tensorflow_text==2.2.0 \
45
58
tensorflow-graphics-gpu==1.0.0 \
46
59
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 \
55
64
tensorflow-lattice==2.0.4 \
56
65
tensorflow_ranking==0.3.1 \
57
66
tensorflow-compression==1.3 \
58
67
tensor2tensor==1.15.6 \
59
68
mesh-tensorflow==0.1.13 \
60
- tensorflow-io==0.13.0 \
61
- jupyter-tensorboard==0.2.0 \
62
69
tensorwatch==0.9.1 \
70
+ tensorflow-io==0.13.0 \
63
71
jupyterlab-nvdashboard==0.3.1
64
72
65
73
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}
76
84
RUN ldconfig /usr/local/cuda/targets/x86_64-linux/lib/stubs && \
77
85
HOROVOD_GPU_ALLREDUCE=NCCL HOROVOD_GPU_BROADCAST=NCCL \
78
86
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 && \
80
88
ldconfig
81
89
82
90
RUN python3 -m pip install --no-cache-dir --upgrade \
83
91
mpi4py==3.0.3 \
84
- nni==1.6 \
85
- scikit-nni==0.2.1
92
+ nni==1.6
86
93
87
94
# Install ipython kernelspec
88
95
Run python3 -m ipykernel install --display-name "Lablup FF 20.06 on Python 3.6 (CUDA 10.1)" && \
0 commit comments