Update to tensorflow 2.16.1

Author: snowzach

Dockerfile.aarch64

@@ -20,4 +20,4 @@ RUN apt-get update && apt-get install -y --no-install-recommends \
 RUN python3 -m pip install --no-cache-dir --break-system-packages \
     protobuf==3.20.3 \
     tensorflow-io==0.35.0 \
-    tensorflow==2.14.0
+    tensorflow==2.16.1

Dockerfile.amd64-noavx

@@ -18,5 +18,5 @@ RUN apt-get update && apt-get install -y --no-install-recommends \
 
 # Install Tensorflow NOAVX and Deps
 RUN python3 -m pip install --no-cache-dir --break-system-packages \
-    https://github.com/snowzach/tensorflow-multiarch/releases/download/v2.14.0/tensorflow-2.14.0-cp311-cp311-linux_x86_64.whl
+    https://github.com/snowzach/tensorflow-multiarch/releases/download/v2.16.1/tensorflow-2.16.1-cp311-cp311-linux_x86_64.whl
 

README.md

@@ -2,7 +2,7 @@
 
 This is my attempt to build a recent Tensorflow wheel and Docker image compatible with multiple architectures and processors. 
 
-It's currently using Tensorflow v2.14.0
+It's currently using Tensorflow v2.16.1
 
 It supports the following architectures:
 - ARM32
@@ -16,7 +16,7 @@ The docker images are `docker.io/snowzach/tensorflow-multiarch` [here](https://h
 The default/latest docker images includes arm32, arm64 and amd64-noavx
 
 ## Compiling Tensorflow Wheel
-I checked out the Tensorflow source at version 2.14.0 and I had to edit the file `tensorflow/tools/ci_build/pi/build_raspberry_pi.sh` and insert on line 99 to add an ARMHF target.
+I checked out the Tensorflow source at version 2.16.1 and I had to edit the file `tensorflow/tools/ci_build/pi/build_raspberry_pi.sh` and insert on line 99 to add an ARMHF target.
 ```
 elif [[ $1 == "ARMHF" ]]; then
   PI_COPTS="--config=elinux_armhf
@@ -35,7 +35,7 @@ $ tensorflow/tools/ci_build/ci_build.sh PI-PYTHON38 \
 ```
 After running the command there was working python whl and library in the `output-artifacts` directory.
 
-I renamed the armhf output wheel from `tensorflow-2.14.0-cp38-none-linux_armhf.whl` to `tensorflow-2.14.0-cp38-none-linux_armv7l.whl` to ensure it matched the hardware architecture and would be installed by python.
+I renamed the armhf output wheel from `tensorflow-2.16.1-cp38-none-linux_armhf.whl` to `tensorflow-2.16.1-cp38-none-linux_armv7l.whl` to ensure it matched the hardware architecture and would be installed by python.
 
 # Tensorflow IO
 Tensorflow IO is required in order to install the Tensorflow wheel. There seems to be a chicken-egg problem 
@@ -50,7 +50,7 @@ $ git clone https://github.com/tensorflow/io.git
 $ cd io
 $ git checkout v0.23.1
 $ python3 setup.py install # Install just the python code
-$ python3 -m pip install tensorflow-2.14.0-cp38-none-linux_<arch>.whl # Install tensorflow wheel
+$ python3 -m pip install tensorflow-2.16.1-cp38-none-linux_<arch>.whl # Install tensorflow wheel
 ```
 
 From there I compiled Tensorflow IO with
@@ -74,18 +74,18 @@ I build them with docker buildx with these commands:
 docker run --rm --privileged multiarch/qemu-user-static --reset -p yes
 
 # Build the docker images
-docker buildx build --push --platform linux/arm/v7 -f Dockerfile.armv7l --tag docker.io/snowzach/tensorflow-multiarch:2.14.0-armv7l .
-docker buildx build --push --platform linux/arm64/v8 -f Dockerfile.aarch64 --tag docker.io/snowzach/tensorflow-multiarch:2.14.0-aarch64 .
-docker buildx build --push --platform linux/amd64 -f Dockerfile.amd64-noavx --tag docker.io/snowzach/tensorflow-multiarch:2.14.0-amd64-noavx .
+docker buildx build --push --platform linux/arm/v7 -f Dockerfile.armv7l --tag docker.io/snowzach/tensorflow-multiarch:2.16.1-armv7l .
+docker buildx build --push --platform linux/arm64/v8 -f Dockerfile.aarch64 --tag docker.io/snowzach/tensorflow-multiarch:2.16.1-aarch64 .
+docker buildx build --push --platform linux/amd64 -f Dockerfile.amd64-noavx --tag docker.io/snowzach/tensorflow-multiarch:2.16.1-amd64-noavx .
 
-# Tag the releases as 2.14.0
-docker manifest push --purge docker.io/snowzach/tensorflow-multiarch:2.14.0
-docker manifest create docker.io/snowzach/tensorflow-multiarch:2.14.0 docker.io/snowzach/tensorflow-multiarch:2.14.0-armv7l docker.io/snowzach/tensorflow-multiarch:2.14.0-aarch64 docker.io/snowzach/tensorflow-multiarch:2.14.0-amd64-noavx
-docker manifest push docker.io/snowzach/tensorflow-multiarch:2.14.0
+# Tag the releases as 2.16.1
+docker manifest push --purge docker.io/snowzach/tensorflow-multiarch:2.16.1
+docker manifest create docker.io/snowzach/tensorflow-multiarch:2.16.1 docker.io/snowzach/tensorflow-multiarch:2.16.1-armv7l docker.io/snowzach/tensorflow-multiarch:2.16.1-aarch64 docker.io/snowzach/tensorflow-multiarch:2.16.1-amd64-noavx
+docker manifest push docker.io/snowzach/tensorflow-multiarch:2.16.1
 
 # Tag the releases as latest
 docker manifest push --purge docker.io/snowzach/tensorflow-multiarch:latest
-docker manifest create docker.io/snowzach/tensorflow-multiarch:latest docker.io/snowzach/tensorflow-multiarch:2.14.0-armv7l docker.io/snowzach/tensorflow-multiarch:2.14.0-aarch64 docker.io/snowzach/tensorflow-multiarch:2.14.0-amd64-noavx
+docker manifest create docker.io/snowzach/tensorflow-multiarch:latest docker.io/snowzach/tensorflow-multiarch:2.16.1-armv7l docker.io/snowzach/tensorflow-multiarch:2.16.1-aarch64 docker.io/snowzach/tensorflow-multiarch:2.16.1-amd64-noavx
 docker manifest push docker.io/snowzach/tensorflow-multiarch:latest
 
 ```

build-tensorflow-noavx.sh

@@ -12,8 +12,8 @@ apt-get update && apt-get install -y --no-install-recommends \
     libc6-dev libstdc++6 libusb-1.0-0 patchelf llvm-16 clang-16
 
 export PROTOC_VERSION="3.20.3"
-export BAZEL_VERSION="6.1.0"
-export TF_VERSION="v2.14.0"
+export BAZEL_VERSION="6.5.0"
+export TF_VERSION="v2.16.1"
 export TF_PYTHON_VERSION=3.11
 
 # Install protoc