【AutoKernel系列教程】一、开发环境快速入门 - 极术社区

文章转载于: Tengine开发者社区
作者:小O妹

AutoKernel

系列教程

- 开发环境快速入门 -

在此教程中，我们将会介绍如何安装/配置AutoKernel的开发环境，并介绍本项目所依赖的两大组件Tengine, Halide。为了减少开发者配置环境的遇到的问题，目前我们提供了Docker来配置所需的基本环境。后续AutoKernel团队会提供更多的环境配置方式。

AutoKernel 安装指引

AutoKernel提供了docker镜像，提供了AutoKernel的开发环境。（如果你还没安装docker,请自行查看docker的官方安装文档）

1、拉取镜像(可能需要一段时间，请耐心等待，根据网速，可能需要10-20mins)

docker pull openailab/autokernel

2、创建容器，进入开发环境

docker run -ti openailab/autokernel /bin/bash

进入到docker容器里

root@39bfb5ea515d:/workspace#

*注意，如果你已经创建了容器，那你只需要启动容器，并且进入即可。否则，你之前的改动不会在新创建的容器中生效。

查看之前创建的容器, 你可以通过命令docker container rename 来重命名你的容器，这里，我们的容器叫做autokernel

$ docker container ls -a

启动容器

docker start autokernel

进入容器

docker exec -ti autokernel /bin/bash

3、docker里面已经安装好Halide, Tengine

/workspace/Halide  # Halide

4、克隆AutoKernel项目

git clone https://github.com/OAID/AutoKernel.git

至此，我们后面所需的环境文件都已经准备完毕。

Halide

Halide 是一个DSL编程语言，他将算法和硬件后端分离了。本项目将使用Halide的DSL 以及IR。docker里面已经安装好Halide，并且配置好了Python的API。

Halide相关的文件都在/workspace/Halide/文件夹下，Halide的安装文件都在/workspace/Halide/halide-build 文件夹下。

cd /workspace/Halide/halide-build

Halide相关头文件在/workspace/Halide/halide-build/include

root@bd3faab0f079:/workspace/Halide/halide-build/include# ls

编译好的Halide库在/workspace/Halide/halide-build/src目录下, 可以看到libHalide.so

root@bd3faab0f079:/workspace/Halide/halide-build/src# ls

运行Halide小程序

cd /workspace/Halide/halide-build

运行结果

Success!

运行Halide的Python接口
首先查看Python的系统路径

python

可以看到Python的系统路径已经有Halide的编译后的python包路径'/workspace/Halide/halide-build/python\_bindings/src'

python

直接import halide成功！

Tengine

Tengine是一个轻量级高性能深度神经网络推理引擎。本项目将基于Tengine进行算子开发优化的工作。

docker里面已经安装好Tengine，相关文件都在/workspace/Tengine/目录下

cd /workspace/Tengine/build

Tengine相关头文件在/workspace/Tengine/build/install/include

root@bd3faab0f079:/workspace/Tengine/build/install/include# ls

编译好的Tengine库在/workspace/Tengine/build/install/lib目录下, 可以看到libtengine-lite.so

root@bd3faab0f079:/workspace/Tengine/build/install/lib# ls

运行Tengine小程序

该示例跑了Tengine在目标电脑上各个网络模型的性能benchmark

cd /workspace/Tengine/benchmark

运行结果

start to run register cpu allocator

TengineGitHub开源链接:https://github.com/OAID/Tengine
欢迎大家Star、Watch、Fork三部曲

推荐阅读

更多Tengine相关内容请关注Tengine-边缘AI推理框架专栏。