tvm python 安装

参考链接：https://blog.csdn.net/sanallen/article/details/81430150

安装LLVM
如下图所示TVM对于ARM后端的支持是基于LLVM编译器实现的，TVM对于的LLVM版本的最低要求为4.0，最高版本不能超过6.0。本次安装LLVM版本6.0。

添加LLVM相关的apt-get源
第一步，首先编辑 /etc/apt/sources.list,加入以下源：

deb http://apt.llvm.org/xenial/ llvm-toolchain-xenial-6.0 main
deb-src http://apt.llvm.org/xenial/ llvm-toolchain-xenial-6.0 main     

添加源后务必运行apt-get update,如果有错误提示，先运行第二步，然后apt-get update）

第二步，安装前必须取得相关证书

wget -O - http://llvm.org/apt/llvm-snapshot.gpg.key|sudo apt-key add -
安装llvm
apt-get install clang-6.0 lldb-6.0
显示如下即安装成功

上面的脚本只适用于Ubuntu16.04添加LLVM 6.0，如果你使用的是其他的操作系统和LLVM版本可以参考这里。

 

编译TVM(host版本)

安装cmake

由于TVM 0.4.0版本首次采用cmake作为编译配置工具，需要安装cmake。

sudo apt-get install cmake

下载TVM
# git命令中选项recursive表示要将项目全部文件都下载，包括子模块。
git clone --recursive https://github.com/dmlc/tvm/
apt-get update
apt-get install -y python python-dev python-setuptools gcc libtinfo-dev zlib1g-dev

配置LLVM选项
在运行cmake-gui命令之前，修改tvm/cmake/config.cmake内容

set(USE_LLVM ON）

编译TVM

完成上述修改后，执行如下命令进行编译选项配置：

cd tvm
mkdir build
cd tvm/build
cmake-gui

• 找到 LLVM设置成on，GRAPH_RUNTIME_DEBUG 设置成on

 

 

make -j4

如果编译正常完成之后，会在tvm的build目录下面生成一些库文件。

• libnnvm_compiler.so
• libtvm_runtime.so
• libtvm.so
• libtvm_topi.so

设置PYTHONPATH环境变量

在安装之前先在/etc/bash.bashrc中设置PYTHONPATH环境变量

export TVM_HOME=/path/to/tvm
export PYTHONPATH=$TVM_HOME/python:$TVM_HOME/topi/python:$TVM_HOME/nnvm/python:${PYTHONPATH}

生效bashrc

source ~/.bashrc