Use TensorFlow 2.16.1

Dobiasd · Apr 4, 2024 · da9232b · da9232b
1 parent 5637438
commit da9232b
Show file tree

Hide file tree

Showing 3 changed files with 4 additions and 4 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.15.0"
+          sudo pip3 install numpy scipy h5py "tensorflow==2.16.1"
           echo "Version numbers of TensorFlow and Keras:"
           python3 -c "import tensorflow as tf; import tensorflow.keras; print(tf.__version__)"
           # FunctionalPlus

diff --git a/README.md b/README.md
@@ -139,7 +139,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.15.0 (These are the tested versions, but somewhat older ones might work too.)
+- TensorFlow 2.16.1 (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,13 +20,13 @@ 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.15.0' --single-branch --depth 1 https://github.com/tensorflow/tensorflow.git
+RUN git clone -b 'v2.16.1' --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
 RUN apt-get install -y patchelf
 RUN ./bazel-bin/tensorflow/tools/pip_package/build_pip_package /tmp/tensorflow_pkg
-RUN pip install /tmp/tensorflow_pkg/tensorflow-2.15.0-cp310-cp310-linux_x86_64.whl
+RUN pip install /tmp/tensorflow_pkg/tensorflow-2.16.1-cp310-cp310-linux_x86_64.whl
 WORKDIR /
 
 RUN git clone -b 'v0.2.23' --single-branch --depth 1 https://github.com/Dobiasd/FunctionalPlus && cd FunctionalPlus && mkdir -p build && cd build && cmake .. && make && make install