1
1
FROM lablup/common-base:py38-cuda11.1
2
2
3
3
RUN python3 -m pip install --no-cache-dir --upgrade \
4
- -q git+https://github.com/google-research/tensorflow_constrained_optimization
5
-
4
+ -q git+https://github.com/google-research/tensorflow_constrained_optimization
5
+
6
6
RUN python3 -m pip install --no-cache-dir \
7
7
--use-feature=2020-resolver \
8
8
requests_oauthlib==1.2.0 \
@@ -21,18 +21,9 @@ RUN python3 -m pip install --no-cache-dir \
21
21
tfx==0.25.0 \
22
22
tfx-bsl==0.25.0 \
23
23
tf-agents==0.6.0
24
+
24
25
RUN python3 -m pip install --no-cache-dir \
25
26
--use-feature=2020-resolver \
26
- tensorflow-cloud==0.1.10 \
27
- tensorflow-hub==0.9.0 \
28
- tensorflow_text==2.4.1 \
29
- tensorflow-graphics-gpu==1.0.0 \
30
- tensorflow-gan==2.0.0 \
31
- tensorflow-data-validation \
32
- tensorflow-plot==0.3.2 \
33
- tensorflow-lattice==2.0.7 \
34
- tensorflow_ranking==0.3.2 \
35
- # tensorflow-compression==1.3 \
36
27
mesh-tensorflow==0.1.13 \
37
28
google_cloud_spanner==1.19.0 \
38
29
pymongo==3.11.2 \
@@ -50,7 +41,22 @@ RUN python3 -m pip install --no-cache-dir \
50
41
proto-plus==1.13.0 \
51
42
oauth2client==4.1.3 \
52
43
joblib==0.14.0 \
53
- libcst==0.3.14 \
44
+ libcst==0.3.14
45
+
46
+ RUN python3 -m pip install --no-cache-dir \
47
+ --use-feature=2020-resolver \
48
+ tensorflow-cloud==0.1.10 \
49
+ tensorflow-hub==0.9.0 \
50
+ tensorflow_text==2.4.1 \
51
+ tensorflow-graphics-gpu==1.0.0 \
52
+ tensorflow-gan==2.0.0
53
+ RUN python3 -m pip install --no-cache-dir \
54
+ --use-feature=2020-resolver \
55
+ tensorflow-data-validation \
56
+ tensorflow-plot==0.3.2 \
57
+ # tensorflow-lattice==2.0.7 \
58
+ tensorflow_ranking==0.3.2 \
59
+ # tensorflow-compression==1.3 \
54
60
tensorflow-io==0.17.0
55
61
56
62
RUN python3 -m pip install --extra-index-url \
@@ -64,7 +70,7 @@ RUN ldconfig /usr/local/cuda/targets/x86_64-linux/lib/stubs && \
64
70
HOROVOD_GPU_ALLREDUCE=NCCL HOROVOD_GPU_BROADCAST=NCCL HOROVOD_NCCL_LINK=SHARED \
65
71
HOROVOD_WITH_TENSORFLOW=1 HOROVOD_WITHOUT_PYTORCH=1 HOROVOD_WITHOUT_MXNET=1\
66
72
HOROVOD_GPU=CUDA \
67
- python3 -m pip install --no-cache-dir horovod==0.21.0 && \
73
+ python3 -m pip install --no-cache-dir horovod==0.21.1 && \
68
74
ldconfig
69
75
70
76
RUN python3 -m pip install --no-cache-dir \
@@ -88,7 +94,7 @@ LABEL ai.backend.kernelspec="1" \
88
94
ai.backend.resource.min.cuda.shares=0 \
89
95
ai.backend.runtime-type="python" \
90
96
ai.backend.runtime-path="/usr/bin/python3" \
91
- ai.backend.service-ports="ipython:pty:3000,jupyter:http:8080 ,jupyterlab:http:8090,vscode:http:8180,tensorboard:http:6006,mlflow-ui:preopen:5000,nniboard:preopen:8080"
97
+ ai.backend.service-ports="ipython:pty:3000,jupyter:http:8081 ,jupyterlab:http:8090,vscode:http:8180,tensorboard:http:6006,mlflow-ui:preopen:5000,nniboard:preopen:8080"
92
98
93
99
WORKDIR /home/work
94
100
# vim: ft=dockerfile
0 commit comments