Skip to content

Commit e0aefba

Browse files
committed
fixed tensorboard debug port
1 parent 85701da commit e0aefba

File tree

1 file changed

+21
-15
lines changed

1 file changed

+21
-15
lines changed

python-tensorflow/Dockerfile.2.4-py38-cuda11.1

Lines changed: 21 additions & 15 deletions
Original file line numberDiff line numberDiff line change
@@ -1,8 +1,8 @@
11
FROM lablup/common-base:py38-cuda11.1
22

33
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+
66
RUN python3 -m pip install --no-cache-dir \
77
--use-feature=2020-resolver \
88
requests_oauthlib==1.2.0 \
@@ -21,18 +21,9 @@ RUN python3 -m pip install --no-cache-dir \
2121
tfx==0.25.0 \
2222
tfx-bsl==0.25.0 \
2323
tf-agents==0.6.0
24+
2425
RUN python3 -m pip install --no-cache-dir \
2526
--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 \
3627
mesh-tensorflow==0.1.13 \
3728
google_cloud_spanner==1.19.0 \
3829
pymongo==3.11.2 \
@@ -50,7 +41,22 @@ RUN python3 -m pip install --no-cache-dir \
5041
proto-plus==1.13.0 \
5142
oauth2client==4.1.3 \
5243
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 \
5460
tensorflow-io==0.17.0
5561

5662
RUN python3 -m pip install --extra-index-url \
@@ -64,7 +70,7 @@ RUN ldconfig /usr/local/cuda/targets/x86_64-linux/lib/stubs && \
6470
HOROVOD_GPU_ALLREDUCE=NCCL HOROVOD_GPU_BROADCAST=NCCL HOROVOD_NCCL_LINK=SHARED \
6571
HOROVOD_WITH_TENSORFLOW=1 HOROVOD_WITHOUT_PYTORCH=1 HOROVOD_WITHOUT_MXNET=1\
6672
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 && \
6874
ldconfig
6975

7076
RUN python3 -m pip install --no-cache-dir \
@@ -88,7 +94,7 @@ LABEL ai.backend.kernelspec="1" \
8894
ai.backend.resource.min.cuda.shares=0 \
8995
ai.backend.runtime-type="python" \
9096
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"
9298

9399
WORKDIR /home/work
94100
# vim: ft=dockerfile

0 commit comments

Comments
 (0)