diff --git a/.github/workflows/ci.yml b/.github/workflows/ci.yml index a84da850..5ce6d96e 100644 --- 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 index bbf59b1d..71d71ffd 100644 --- 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 index f5f9d65f..e7b495b4 100644 --- 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