Skip to content

Commit

Permalink
[CI] Use pre-built libtensorflowlite_c.so only and use the official l…
Browse files Browse the repository at this point in the history
…ibtensorflow repository
  • Loading branch information
hydai committed Dec 25, 2020
1 parent f243a48 commit a8aed2a
Show file tree
Hide file tree
Showing 2 changed files with 10 additions and 6 deletions.
14 changes: 9 additions & 5 deletions README.md
Original file line number Diff line number Diff line change
Expand Up @@ -25,16 +25,20 @@ For the tensorflow extensions:

* libpng-dev
* libjpeg-dev
* libtensorflow.so
* Please refer to the TensorFlow website for more details
```base
RUN wget https://storage.googleapis.com/tensorflow/libtensorflow/libtensorflow-cpu-linux-x86_64-2.4.0.tar.gz \
&& tar -C /usr/local -xzf libtensorflow-cpu-linux-x86_64-2.4.0.tar.gz \
&& ldconfig
```
* libtensorflowlite\_c.so
* You can use the following commands to get libtensorflowlite\_c.so
```bash
# Download TensorFlow deps
wget https://github.com/second-state/ssvm-tensorflow-deps/releases/download/0.1.0/ssvm-tensorflow-deps-0.1.0-linux-x64.tar.gz
tar -zxvf ssvm-tensorflow-deps-0.1.0-linux-x64.tar.gz
rm -f ssvm-tensorflow-deps-0.1.0-linux-x64.tar.gz
wget https://github.com/second-state/ssvm-tensorflow-deps/releases/download/0.1.0/ssvm-tensorflow-deps-lite-0.1.0-linux-x64.tar.gz
tar -zxvf ssvm-tensorflow-deps-lite-0.1.0-linux-x64.tar.gz
cp ./libtensorflowlite_c.so /usr/local/lib
rm ./*.so
rm ./*.so.*
```

## Prepare environment
Expand Down
2 changes: 1 addition & 1 deletion utils/docker/Dockerfile
Original file line number Diff line number Diff line change
Expand Up @@ -2,7 +2,7 @@ FROM secondstate/ssvm:latest

ARG DEBIAN_FRONTEND=noninteractive
ARG TENSORFLOW_TARBALL=libtensorflow-cpu-linux-x86_64-2.4.0.tar.gz
ARG TENSORFLOW_LITE_TARBALL=ssvm-tensorflow-deps-0.1.0-linux-x64.tar.gz
ARG TENSORFLOW_LITE_TARBALL=ssvm-tensorflow-deps-lite-0.1.0-linux-x64.tar.gz
ARG SSVM_TENSORFLOW_DEPS_VERSION=0.1.0
ENV RUSTUP_HOME=/usr/local/rustup \
CARGO_HOME=/usr/local/cargo \
Expand Down

0 comments on commit a8aed2a

Please sign in to comment.