TVM个人实验(一)--TVM在ubuntu下安装

wcj0626

已于 2023-04-26 10:17:59 修改

阅读量1.1k

点赞数 1

CC 4.0 BY-SA版权

分类专栏： TVM 文章标签： ubuntu TVM

于 2023-04-26 10:16:26 首次发布

本文链接：https://blog.youkuaiyun.com/wcj0626/article/details/130379852

TVM 专栏收录该内容

1 篇文章

订阅专栏

本文记录了在Ubuntu系统中安装TVM的过程，特别是遇到的LLVM版本匹配问题。作者首先介绍了安装LLVM和Clang的步骤，然后强调了在配置TVM的CMake文件时，需指定使用llvm-config-14以适配已安装的LLVM14.0.0版本。如果不正确设置，会导致运行示例时出现错误。文章提供了解决此特定错误的方法。

摘要生成于 C知道，由 DeepSeek-R1 满血版支持，前往体验 >

从今天开始，开始研究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遇到的坑。