Linux系统下安装TVM

Table of Contents

安装环境

  • Linux Mint
  • gcc/g++ 9.4
  • git
  • python 3.8
  • cmake
  • ninja
  • llvm 10.0.0
  • tvm 0.8
  • 额外依赖包

sudo apt install libtinfo-dev zlib1g-dev build-essential libedit-dev libxml2-dev

LLVM安装

预编译版本-PreBuild

预编译版本无需编译，仅支持llvm和clang，没有其他插件，直接官网下载。

源码编译安装

源码编译可以编译除llvm和clang外的工具，但是编译过程中对环境要求比较严格，容易出现各种错误，本文尝试各种方法后使用该方法编译成功。

  • 从GitHub下载源码

    git clone -b 分支 https://github.com/llvm/llvm-project.git

  • 进入源码文件夹创建build文件并打开，输入以下命令生成配置文件。 

    cmake -G Ninja -DCMAKE_INSTALL_PREFIX=/path/to/llvm-build -DCMAKE_BUILD_TYPE=Release  -DLLVM_TARGETS_TO_BUILD=X86 -DLLVM_ENABLE_PROJECTS='libcxx;libcxxabi;clang;lldb' ../llvm

    -DLLVM_ENABLE_PROJECTS 选项用于选择开启的插件。

  • 使用Ninja编译llvm。

    ninja -j12 -j开启多核编译。

  • 配置环境变量
    • PATH指向bin文件夹
    • library指向lib文件夹

TVM安装

  • 从GitHub下载源码

    git clone --recursive https://github.com/apache/tvm tvm

  • 编辑tvm文件夹下的 CMakefile.txt 文件

    设置需要开启的选项为ON。

  • 创建build文件夹并进入

    mkdir build cd build

  • 编译文件 

    cmake -G Ninja -DCMAKE_BUILD_TYPE=Debug .. 
ninja -j12

  • 配置环境变量 

    TVM_HOME=/path/to/tvm  
PYTHONPATH=$TVM_HOME/python:${PYTHONPATH}

Author: Frank

Created: 2021-04-15 周四 01:29

Validate