@@ -34,7 +34,7 @@ RUN ln -s /usr/local/cuda/lib64/stubs/libcuda.so /usr/local/cuda/lib64/stubs/lib
34
34
{{ end }}
35
35
36
36
# Keep these variables in sync if base image is updated.
37
- ENV TENSORFLOW_VERSION=2.9.1
37
+ ENV TENSORFLOW_VERSION=2.9.2
38
38
39
39
# We need to redefine the ARG here to get the ARG value defined above the FROM instruction.
40
40
# See: https://docs.docker.com/engine/reference/builder/#understand-how-arg-and-from-interact
@@ -177,7 +177,7 @@ RUN pip install pysal && \
177
177
# Use `conda install -c h2oai h2o` once Python 3.7 version is released to conda.
178
178
apt-get install -y default-jre-headless && \
179
179
pip install -f https://h2o-release.s3.amazonaws.com/h2o/latest_stable_Py.html h2o && \
180
- pip install tensorflow-gcs-config== ${TENSORFLOW_VERSION} && \
180
+ pip install " tensorflow-gcs-config<= ${TENSORFLOW_VERSION}" && \
181
181
# TODO(b/207851560) Upgrade to 0.17.1 once the base image with TensorFlow 2.9.1 is out.
182
182
pip install tensorflow-addons==0.17.1 && \
183
183
pip install tensorflow_decision_forests==0.2.7 && \
@@ -496,7 +496,8 @@ RUN pip install flashtext && \
496
496
pip install optuna && \
497
497
pip install plotly_express && \
498
498
pip install albumentations && \
499
- pip install catalyst && \
499
+ # Breaks protobuf compatibiilty in newer versions:
500
+ pip install catalyst tensorboardX==2.5.1 && \
500
501
# b/206990323 osmx 1.1.2 requires numpy >= 1.21 which we don't want.
501
502
pip install osmnx==1.1.1 && \
502
503
apt-get -y install libspatialindex-dev && \
@@ -513,7 +514,8 @@ RUN pip install flashtext && \
513
514
#pip install dlib && \
514
515
pip install kaggle-environments && \
515
516
pip install geopandas && \
516
- pip install nnabla && \
517
+ # Breaks protobuf compatibiilty:
518
+ #pip install nnabla && \
517
519
pip install vowpalwabbit && \
518
520
pip install pydub && \
519
521
pip install pydegensac && \
0 commit comments