Skip to content

Commit

Permalink
[DOCKER] Tensorflow upgrade to 2.18.0
Browse files Browse the repository at this point in the history
upgrade TF to latest version and fixup tflite build issue accordingly.
  • Loading branch information
srkreddy1238 committed Jan 30, 2025
1 parent 2340081 commit 4073a56
Show file tree
Hide file tree
Showing 3 changed files with 10 additions and 5 deletions.
4 changes: 2 additions & 2 deletions docker/install/ubuntu_install_tensorflow.sh
Original file line number Diff line number Diff line change
Expand Up @@ -21,5 +21,5 @@ set -u
set -o pipefail

pip3 install \
keras==2.9 \
tensorflow==2.9.1
keras==3.5.0 \
tensorflow==2.18.0
9 changes: 7 additions & 2 deletions docker/install/ubuntu_install_tflite.sh
Original file line number Diff line number Diff line change
Expand Up @@ -26,7 +26,7 @@ set -o pipefail
TENSORFLOW_VERSION=$(python3 -c "import tensorflow; print(tensorflow.__version__)" 2> /dev/null)

# Download, build and install flatbuffers
git clone --branch=v1.12.0 --depth=1 --recursive https://github.com/google/flatbuffers.git
git clone --branch=v25.1.24 --depth=1 --recursive https://github.com/google/flatbuffers.git
cd flatbuffers
cmake -G "Unix Makefiles" -DCMAKE_BUILD_TYPE=Release -DCMAKE_CXX_FLAGS="-Wno-class-memaccess"
make install -j8
Expand All @@ -52,7 +52,12 @@ cd -

# Setup tflite from schema
mkdir tflite
cp tensorflow/tensorflow/lite/schema/schema.fbs tflite
if [ -f tensorflow/tensorflow/compiler/mlir/lite/schema/schema.fbs ] ; then
cp tensorflow/tensorflow/compiler/mlir/lite/schema/schema.fbs tflite
else
cp tensorflow/tensorflow/lite/schema/schema.fbs tflite
fi

cd tflite
flatc --python schema.fbs

Expand Down
2 changes: 1 addition & 1 deletion docker/install/ubuntu_install_wasmtime.sh
Original file line number Diff line number Diff line change
Expand Up @@ -23,7 +23,7 @@ apt-install-and-clear -y --no-install-recommends libc6-dev-i386
export WASMTIME_HOME=/opt/wasmtime
curl https://wasmtime.dev/install.sh -sSf | bash
export PATH="${WASMTIME_HOME}/bin:${PATH}"
rustup target add wasm32-wasi
rustup target add wasm32-wasip1

# make rust usable by all users after install during container build
chmod -R a+rw /opt/rust

0 comments on commit 4073a56

Please sign in to comment.