Update to TensorFlow 2.17

Dobiasd · Jul 13, 2024 · 199a604 · 199a604
1 parent 3fb55bb
commit 199a604
Show file tree

Hide file tree

Showing 3 changed files with 3 additions and 3 deletions.
diff --git a/.github/workflows/ci.yml b/.github/workflows/ci.yml
@@ -13,7 +13,7 @@ jobs:
           sudo apt-get install libblas-dev liblapack-dev libatlas-base-dev gfortran
           # python libs
           sudo pip3 install --upgrade pip
-          sudo pip3 install numpy scipy h5py "tensorflow==2.16.1"
+          sudo pip3 install numpy scipy h5py "tensorflow==2.17"
           echo "Python version:"
           python3 --version
           echo "Version numbers of TensorFlow and Keras:"

diff --git a/README.md b/README.md
@@ -140,7 +140,7 @@ Requirements and Installation
 
 - A **C++14**-compatible compiler: Compilers from these versions on are fine: GCC 4.9, Clang 3.7 (libc++ 3.7) and Visual C++ 2015
 - Python 3.7 or higher
-- TensorFlow 2.16.1 (These are the tested versions, but somewhat older ones might work too.)
+- TensorFlow 2.17 (These are the tested versions, but somewhat older ones might work too.)
 
 Guides for different ways to install frugally-deep can be found in [`INSTALL.md`](INSTALL.md).
 

diff --git a/test/Dockerfile b/test/Dockerfile
@@ -20,7 +20,7 @@ RUN echo "deb [arch=amd64] https://storage.googleapis.com/bazel-apt stable jdk1.
 RUN apt-get update && apt-get install -y bazel-5.1.1
 RUN ln -s /usr/bin/bazel-5.1.1 /usr/bin/bazel
 
-RUN git clone -b 'v2.16.1' --single-branch --depth 1 https://github.com/tensorflow/tensorflow.git
+RUN git clone -b 'v2.17' --single-branch --depth 1 https://github.com/tensorflow/tensorflow.git
 WORKDIR /tensorflow
 RUN ./configure
 RUN bazel build --jobs 3 --local_ram_resources=HOST_RAM*.3 -c opt //tensorflow/tools/pip_package:build_pip_package