提示:仅记录一下本次成功安装的过程用于参考
目录
前言
记录一下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 文件:
-
如果要时候用某些后端和库(如CUDA, OpenCL, RCOM, METAL, VULKAN,等)。这里以CUDA为例,将 set(USE_CUDA OFF) 修改为 set(USE_CUDA ON),其他库同理。
-
为了帮助调试,需要启用嵌入式图形执行器和调试功能
set(USE_GRAPH_DEBUG ON)
、set(USE_PROFILER ON)
-
使用 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