Skip to content

Commit

Permalink
🐳update Dockfile
Browse files Browse the repository at this point in the history
  • Loading branch information
dong wang committed Aug 28, 2023
1 parent 11897a3 commit 5fe1c8c
Show file tree
Hide file tree
Showing 3 changed files with 50 additions and 3 deletions.
30 changes: 30 additions & 0 deletions Dockerfile
Original file line number Diff line number Diff line change
@@ -0,0 +1,30 @@
FROM mister5ive/ai.deploy.box:1.1.0
MAINTAINER "Hulk Wang"

LABEL description="AIDeployBox:A toolbox for deep learning model deployment using C++."

ARG AiDB_HOME=/aidb
ARG LINUX_LIBS=linux.zip
ARG MODEL=models-lite.zip

ARG LIBS_URL=https://github.com/TalkUHulk/ai.deploy.box/releases/download/1.1.0/$LINUX_LIBS
ARG MODEL_LITE_URL=https://github.com/TalkUHulk/ai.deploy.box/releases/download/1.1.0/$MODEL

ADD . $AiDB_HOME/

RUN curl -LJO $LIBS_URL \
&& curl -LJO $MODEL_LITE_URL \
&& unzip $LINUX_LIBS -d $AiDB_HOME/libs/ \
&& rm -f $LINUX_LIBS \
&& unzip $MODEL -d $AiDB_HOME/ \
&& rm -f $MODEL \
&& mv $AiDB_HOME/models-lite $AiDB_HOME/models \
&& mkdir $AiDB_HOME/build \
&& cd $AiDB_HOME/build \
&& cmake -DENGINE_MNN=ON -DENGINE_ORT=ON -DENGINE_NCNN=ON -DENGINE_TNN=ON -DENGINE_OPV=ON -DENGINE_PPLite=ON -DBUILD_SAMPLE=ON ../ \
&& make --jobs=$(nproc --all)

ENV LD_LIBRARY_PATH $AiDB_HOME/libs/linux/x86_64:$AiDB_HOME/libs/linux/x86_64/tnn:$AiDB_HOME/libs:$AiDB_HOME/libs/linux/x86_64/openvino:$AiDB_HOME/libs/linux/x86_64/paddlelite:$LD_LIBRARY_PATH

WORKDIR $AiDB_HOME/

10 changes: 10 additions & 0 deletions README.md
Original file line number Diff line number Diff line change
Expand Up @@ -146,6 +146,16 @@ docker pull mister5ive/ai.deploy.box

### Build

#### Docker(recommend)

```asm
git lfs clone https://github.com/TalkUHulk/ai.deploy.box.git
cd ai.deploy.box.git
docker build -t aidb-dev .
docker run -it --name aidb-test aidb-dev
```

#### From Source code

```asm
git lfs clone https://github.com/TalkUHulk/ai.deploy.box.git
Expand Down
13 changes: 10 additions & 3 deletions README_cn.md
Original file line number Diff line number Diff line change
Expand Up @@ -134,15 +134,23 @@

<details>
<summary>推荐使用docker</summary>

```java
docker pull mister5ive/ai.deploy.box
```
</details>


### Build

#### Docker(推荐)

```asm
git lfs clone https://github.com/TalkUHulk/ai.deploy.box.git
cd ai.deploy.box.git
docker build -t aidb-dev .
docker run -it --name aidb-test aidb-dev
```

#### 源码编译

```asm
git clone https://github.com/TalkUHulk/ai.deploy.box.git
Expand Down Expand Up @@ -674,7 +682,6 @@ A: 使用带fp16标签的库

Q: kernel for 'conv2d' is not supported by Paddle-lite.
A: 转模型--valid_targets =arm, 打开fp16,opt\lib版本对应

5. Android OpenVINO

Q: How to deploy models in android with openvino([reference](https://github.com/openvinotoolkit/openvino/issues/15006)).
Expand Down

0 comments on commit 5fe1c8c

Please sign in to comment.