tvm安装

1、下载源码

git clone --recursive https://github.com/apache/tvm tvm
cd tvm
git submodule init
git submodule update

2、下载依赖

sudo apt-get update
sudo apt-get install -y python3 python3-dev python3-setuptools gcc libtinfo-dev zlib1g-dev build-essential cmake libedit-dev libxml2-dev
mkdir build
cp cmake/config.cmake build

3、安装llvm

去[llvm官网](LLVM Download Page)下载适合自己系统的压缩包，下面以ubuntu20.04为例

wget https://github.com/llvm/llvm-project/releases/download/llvmorg-13.0.0/clang+llvm-13.0.0-x86_64-linux-gnu-ubuntu-20.04.tar.xz
tar xJvf clang+llvm-13.0.0-x86_64-linux-gnu-ubuntu-20.04.tar.xz

编辑~/.bashrc，加入以下内容

export PATH=$PATH:/path/to/clang+llvm-13.0.0-x86_64-linux-gnu-ubuntu-20.04/bin
source ~/.bashrc
#使用下面命令测试llvm是否安装成功
llvm-config --version

4、自定义编译选项

cd tvm
vi build/config.cmake

将其中一行set(USE_LLVM OFF)改为set(USE_LLVM ON)

如果要加入VTA，将其中一行set(USE_VTA_FSIM OFF)改为set(USE_VTA_FSIM ON)

5、编译TVM

cd build
cmake ..
make -j4

6、增加python支持

在~/.bashrc中加入以下几行

export TVM_HOME=/path/to/tvm
export PYTHONPATH=$TVM_HOME/python:$TVM_HOME/topi/python:$TVM_HOME/nnvm/python:$TVM_HOME/vta/python:${PYTHONPATH}
#如果要加入VTA支持，下面一行要加入
export VTA_HW_PATH=$TVM_HOME/3rdparty/vta-hw

保存~/.bashrc后，执行source ~/.bashrc

7、安装python相关依赖

pip3 install numpy decorator attrs
pip3 install tornado
pip3 install tornado psutil xgboost
pip3 install mypy orderedset antlr4-python3-runtime