Skip to content

Commit 3fb2bf7

Browse files
committed
fixed packages dependency of Tensorflow 2.4
1 parent 00db25a commit 3fb2bf7

File tree

1 file changed

+2
-3
lines changed

1 file changed

+2
-3
lines changed

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

Lines changed: 2 additions & 3 deletions
Original file line numberDiff line numberDiff line change
@@ -72,7 +72,6 @@ RUN python3 -m pip install --no-cache-dir \
7272
nni==1.9 \
7373
mlflow==1.12.1 \
7474
scikit-nni==0.2.1
75-
7675

7776
# Install ipython kernelspec
7877
Run python3 -m ipykernel install --display-name "TensorFlow 2.4 on Python 3.6 & CUDA 11ᇂᇂᇂᇂ.1" && \
@@ -89,7 +88,7 @@ LABEL ai.backend.kernelspec="1" \
8988
ai.backend.resource.min.cuda.shares=0 \
9089
ai.backend.runtime-type="python" \
9190
ai.backend.runtime-path="/usr/bin/python3" \
92-
ai.backend.service-ports="ipython:pty:3000,jupyter:http:8080,jupyterlab:http:8090,vscode:http:8180,tensorboard:http:6006"
93-
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"
92+
9493
WORKDIR /home/work
9594
# vim: ft=dockerfile

0 commit comments

Comments
 (0)