Change build to tf 2.16.1

tensorflow · Apr 10, 2024 · 5ea6f5d · 5ea6f5d
1 parent 4d70341
commit 5ea6f5d
Show file tree

Hide file tree

Showing 4 changed files with 9 additions and 10 deletions.
diff --git a/.bazelversion b/.bazelversion
@@ -1 +1 @@
-6.1.0
+6.5.0
diff --git a/WORKSPACE b/WORKSPACE
@@ -93,10 +93,10 @@ switched_rules_by_language(
 
 http_archive(
     name = "org_tensorflow",
-    sha256 = "9cec5acb0ecf2d47b16891f8bc5bc6fbfdffe1700bdadc0d9ebe27ea34f0c220",
-    strip_prefix = "tensorflow-2.15.0",
+    sha256 = "c729e56efc945c6df08efe5c9f5b8b89329c7c91b8f40ad2bb3e13900bd4876d",
+    strip_prefix = "tensorflow-2.16.1",
     urls = [
-        "https://github.com/tensorflow/tensorflow/archive/refs/tags/v2.15.0.tar.gz",
+        "https://github.com/tensorflow/tensorflow/archive/refs/tags/v2.16.1.tar.gz",
     ],
 )
 
@@ -880,10 +880,9 @@ http_archive(
 # Needed for llvm_toolchain and Golang
 http_archive(
     name = "com_grail_bazel_toolchain",
-    sha256 = "9e6065ded4b7453143e1586d6819729a63cd233114b72bf85ff3435367b02c90",
-    strip_prefix = "bazel-toolchain-edd07e96a2ecaa131af9234d6582875d980c0ac7",
+    sha256 = "4b7999c1fa2c3117bb21651e3c155b152e44ae67b2c311214883d4707dbe183f",
+    strip_prefix = "toolchains_llvm-edd07e96a2ecaa131af9234d6582875d980c0ac7",
     urls = [
-        "https://storage.googleapis.com/mirror.tensorflow.org/github.com/grailbio/bazel-toolchain/archive/edd07e96a2ecaa131af9234d6582875d980c0ac7.tar.gz",
         "https://github.com/grailbio/bazel-toolchain/archive/edd07e96a2ecaa131af9234d6582875d980c0ac7.tar.gz",
     ],
 )

diff --git a/tensorflow_io/python/ops/version_ops.py b/tensorflow_io/python/ops/version_ops.py
@@ -14,5 +14,5 @@
 # ==============================================================================
 """version_ops"""
 
-version = "0.36.0"
-require = "tensorflow>=2.15.0,<2.16.0"
+version = "0.37.0"
+require = "tensorflow>=2.16.0,<2.17.0"
diff --git a/tools/docker/build.Dockerfile b/tools/docker/build.Dockerfile
@@ -9,7 +9,7 @@ ARG BAZEL_OPTIMIZATION
 ADD . /opt/io
 WORKDIR /opt/io
 
-RUN python${PYTHON_VERSION} -m pip install tensorflow==${TENSORFLOW_VERSION}
+RUN python${PYTHON_VERSION} -m pip install tensorflow>=${TENSORFLOW_VERSION}
 
 RUN python$PYTHON_VERSION -m pip uninstall -y tensorflow-io-gcs-filesystem