@@ -25,7 +25,7 @@ RUN python3 -m pip install --no-cache-dir \
25
25
RUN python3 -m pip uninstall -y google-api-core
26
26
RUN python3 -m pip install --no-cache-dir \
27
27
--use-feature=2020-resolver \
28
- google_cloud_spanner==3.22.2 \
28
+ google_cloud_spanner==1.19.3 \
29
29
googleapis-common-protos==1.56.4 \
30
30
google_cloud_videointelligence==1.15.0 \
31
31
google_cloud_vision==0.42.0 \
@@ -45,11 +45,11 @@ RUN python3 -m pip install --no-cache-dir \
45
45
-i https://pypi-nightly.tensorflow.org/simple tensorflow-data-validation==1.10.0
46
46
RUN python3 -m pip install --no-cache-dir \
47
47
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 \
50
50
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
53
53
RUN python3 -m pip install --no-cache-dir \
54
54
tensorflow-gpu==2.10.0 \
55
55
tensorboard-plugin-wit \
@@ -60,13 +60,13 @@ RUN ldconfig /usr/local/cuda/targets/x86_64-linux/lib/stubs && \
60
60
HOROVOD_GPU_ALLREDUCE=NCCL HOROVOD_GPU_BROADCAST=NCCL HOROVOD_NCCL_LINK=SHARED \
61
61
HOROVOD_WITH_TENSORFLOW=1 HOROVOD_WITHOUT_PYTORCH=1 HOROVOD_WITHOUT_MXNET=1\
62
62
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 && \
64
64
ldconfig
65
65
66
66
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 \
70
70
scikit-nni==0.2.1
71
71
72
72
# Install ipython kernelspec
0 commit comments