Skip to content

Commit

Permalink
[tensorflow-aarch64]: Add support to extract tf wheel from the container
Browse files Browse the repository at this point in the history
tf wheel is packaged into the container venv to be able to extract from
the host.

Change-Id: I97398e7fbe7839adabf156cac3dc3ee362159695
  • Loading branch information
snadampal authored and nSircombe committed Dec 7, 2021
1 parent 76922f0 commit d17ec56
Showing 1 changed file with 2 additions and 0 deletions.
2 changes: 2 additions & 0 deletions docker/tensorflow-aarch64/scripts/build-tensorflow.sh
Original file line number Diff line number Diff line change
Expand Up @@ -110,10 +110,12 @@ pip install $(ls -tr wheel-TF$TF_VERSION-py$PY_VERSION-$CC/*.whl | tail)
# Install Tensorflow C++ interface
mkdir -p $VENV_DIR/$package/lib
mkdir -p $VENV_DIR/$package/include
mkdir -p $VENV_DIR/$package/wheel
cp ./bazel-bin/tensorflow/libtensorflow* $VENV_DIR/$package/lib
cp -r ./bazel-bin/tensorflow/include $VENV_DIR/$package/
cp -r $VENV_DIR/lib/python$PY_VERSION/site-packages/tensorflow/include/google \
$VENV_DIR/$package/include
cp $(ls -tr wheel-TF$TF_VERSION-py$PY_VERSION-$CC/*.whl | tail) $VENV_DIR/$package/wheel

# Check the Python installation was sucessfull
cd $HOME
Expand Down

0 comments on commit d17ec56

Please sign in to comment.