Skip to content

Commit

Permalink
Upgrade to Tensorflow 2.13
Browse files Browse the repository at this point in the history
http://b/301473408
  • Loading branch information
djherbis committed Sep 21, 2023
1 parent 0fac236 commit 426347e
Show file tree
Hide file tree
Showing 2 changed files with 6 additions and 7 deletions.
7 changes: 3 additions & 4 deletions Dockerfile.tmpl
Original file line number Diff line number Diff line change
Expand Up @@ -36,9 +36,9 @@ RUN ln -s /usr/local/cuda/lib64/stubs/libcuda.so /usr/local/cuda/lib64/stubs/lib
{{ end }}

# Keep these variables in sync if base image is updated.
ENV TENSORFLOW_VERSION=2.12.0
ENV TENSORFLOW_VERSION=2.13.0
# See https://github.com/tensorflow/io#tensorflow-version-compatibility
ENV TENSORFLOW_IO_VERSION=0.32.0
ENV TENSORFLOW_IO_VERSION=0.34.0

# We need to redefine the ARG here to get the ARG value defined above the FROM instruction.
# See: https://docs.docker.com/engine/reference/builder/#understand-how-arg-and-from-interact
Expand Down Expand Up @@ -520,8 +520,7 @@ RUN pip install flashtext \
pyarabic \
pandasql \
tensorflow_hub \
# b/300552705 Remove pin once we upgrade to TensorFlow 2.13
tensorflowjs==4.10 \
tensorflowjs \
jieba \
# ggplot is broken and main repo does not merge and release https://github.com/yhat/ggpy/pull/668
https://github.com/hbasria/ggpy/archive/0.11.5.zip \
Expand Down
6 changes: 3 additions & 3 deletions config.txt
Original file line number Diff line number Diff line change
@@ -1,7 +1,7 @@
BASE_IMAGE_REPO=gcr.io/deeplearning-platform-release
BASE_IMAGE_TAG=m109
CPU_BASE_IMAGE_NAME=tf2-cpu.2-12.py310
GPU_BASE_IMAGE_NAME=tf2-gpu.2-12.py310
BASE_IMAGE_TAG=m111
CPU_BASE_IMAGE_NAME=tf2-cpu.2-13.py310
GPU_BASE_IMAGE_NAME=tf2-gpu.2-13.py310
LIGHTGBM_VERSION=3.3.2
TORCH_VERSION=2.0.0
TORCHAUDIO_VERSION=2.0.1
Expand Down

0 comments on commit 426347e

Please sign in to comment.