从今天开始,开始研究TVM的用法,用系列文章来记录自己实验的点点滴滴。开头第一篇,介绍如何在ubuntu 下安装TVM。
对于TVM的站点对于如何安装TVM已经讲的非常的清楚。但是在这里记录自己掉坑的地方。
1. 安装llvm
sudo apt-get install llvm
sudo apt-get install clang
查看llvm的版本:
$ llvm-as --version
Ubuntu LLVM version 14.0.0
Optimized build.
Default target: x86_64-pc-linux-gnu
Host CPU: tigerlake
LLVM的版本为14.0.0
2. TVM的编译安装
详细的流程,根据官方文档依次操作就行,但是有一个需要注意的地方。
因为本地安装的是LLVM14版本,所以在config.cmake中设置如下:
set(USE_LLVM llvm-config-14)
如果这个设置不对的话,在跑官方的例子将ONMX模型编译到TVM Runtime时,会报如下错误:
module 'tvm.target._ffi_api' has no attribute 'llvm_version_major'
RuntimeError: LLVM version is not available, please check if you built TVM with LLVM
以上便是自己源码安装TVM遇到的坑。