TVM编译与python环境配置

提示：仅记录一下本次成功安装的过程用于参考

---

目录

前言

一、源码下载

二、编译

1.安装依赖

2.安装llvm

3.开始编译 

4.设置python环境

---

前言

记录一下tvm的编译流程与python环境安装

环境：rockpi5b、Debian11\utuntu20.04、miniconda4.12.0、python3.8

一、源码下载

官网下载地址https://tvm.apache.org/download

二、编译

1.安装依赖

安装最小要求的依赖（ubuntu / debian）：

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

2.安装llvm

使用官方安装脚本安装（仅适用于Debian/Ubuntu）：

wget https://apt.llvm.org/llvm.sh
chmod +x llvm.sh
sudo ./llvm.sh 13

3.开始编译 

首先进入到我们刚才下载的 tvm 源码目录下，新建 build 目录，将 cmake 目录下的 config.cmake 拷贝的 build 目录下：

cd tvm
mkdir build
cp cmake/config.cmake build/

然后按需修改 config.cmake 文件：

1. 如果要时候用某些后端和库（如CUDA, OpenCL, RCOM, METAL, VULKAN,等）。这里以CUDA为例，将 set(USE_CUDA OFF) 修改为 set(USE_CUDA ON)，其他库同理。

2. 为了帮助调试，需要启用嵌入式图形执行器和调试功能 set(USE_GRAPH_DEBUG ON) 、 set(USE_PROFILER ON)

3. 使用 LLVM，这里要使用到我们刚才记下的 LLVM 路径了，设置 set(USE_LLVM /usr/bin/llvm-config-13)

config.cmake 修改完成后，我们开始构建 tvm 及相关库:

cd build
cmake ..
make -j4

4.设置python环境

参考官网教程

Install from Source — tvm 0.11.dev0 documentation

• 安装必要依赖

pip3 install numpy decorator attrs

• 如果要使用RPC跟踪器

pip3 install tornado

• 如果要使用auto-tuning模块

pip3 install tornado psutil 'xgboost<1.6.0' cloudpickle

进入tvm/python目录下，通过setup.py脚本安装python的tvm包：

cd python
python setup.py build
python setup.py install