Skip to content

Commit 9c97fd4

Browse files
committed
fixed dependency
1 parent 134a900 commit 9c97fd4

File tree

1 file changed

+9
-9
lines changed

1 file changed

+9
-9
lines changed

python-tensorflow/Dockerfile.2.10-py38-cuda11.3

Lines changed: 9 additions & 9 deletions
Original file line numberDiff line numberDiff line change
@@ -25,7 +25,7 @@ RUN python3 -m pip install --no-cache-dir \
2525
RUN python3 -m pip uninstall -y google-api-core
2626
RUN python3 -m pip install --no-cache-dir \
2727
--use-feature=2020-resolver \
28-
google_cloud_spanner==3.22.2 \
28+
google_cloud_spanner==1.19.3 \
2929
googleapis-common-protos==1.56.4 \
3030
google_cloud_videointelligence==1.15.0 \
3131
google_cloud_vision==0.42.0 \
@@ -45,11 +45,11 @@ RUN python3 -m pip install --no-cache-dir \
4545
-i https://pypi-nightly.tensorflow.org/simple tensorflow-data-validation==1.10.0
4646
RUN python3 -m pip install --no-cache-dir \
4747
tensorflow-graphics-gpu==1.0.0 \
48-
tensorflow-io==0.17.0 \
49-
tensorflow_ranking==0.3.2 \
48+
tensorflow-io==0.27.0 \
49+
tensorflow_ranking==0.5.1 \
5050
tensorflow-plot==0.3.2 \
51-
tensorflow_text==2.4.1 \
52-
tensorflow-gan==2.0.0
51+
tensorflow_text==2.10.0 \
52+
tensorflow-gan==2.1.0
5353
RUN python3 -m pip install --no-cache-dir \
5454
tensorflow-gpu==2.10.0 \
5555
tensorboard-plugin-wit \
@@ -60,13 +60,13 @@ RUN ldconfig /usr/local/cuda/targets/x86_64-linux/lib/stubs && \
6060
HOROVOD_GPU_ALLREDUCE=NCCL HOROVOD_GPU_BROADCAST=NCCL HOROVOD_NCCL_LINK=SHARED \
6161
HOROVOD_WITH_TENSORFLOW=1 HOROVOD_WITHOUT_PYTORCH=1 HOROVOD_WITHOUT_MXNET=1\
6262
HOROVOD_GPU=CUDA \
63-
python3 -m pip install --no-cache-dir horovod==0.23.0 && \
63+
python3 -m pip install --no-cache-dir horovod==0.26.1 && \
6464
ldconfig
6565

6666
RUN python3 -m pip install --no-cache-dir \
67-
mpi4py==3.0.3 \
68-
nni==1.9 \
69-
mlflow==1.12.1 \
67+
mpi4py==3.1.4 \
68+
nni==2.9 \
69+
mlflow==1.30.0 \
7070
scikit-nni==0.2.1
7171

7272
# Install ipython kernelspec

0 commit comments

Comments
 (0)