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

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

  • 0
    点赞
  • 0
    收藏
    觉得还不错? 一键收藏
  • 0
    评论
评论
添加红包

请填写红包祝福语或标题

红包个数最小为10个

红包金额最低5元

当前余额3.43前往充值 >
需支付:10.00
成就一亿技术人!
领取后你会自动成为博主和红包主的粉丝 规则
hope_wisdom
发出的红包
实付
使用余额支付
点击重新获取
扫码支付
钱包余额 0

抵扣说明:

1.余额是钱包充值的虚拟货币,按照1:1的比例进行支付金额的抵扣。
2.余额无法直接购买下载,可以购买VIP、付费专栏及课程。

余额充值