Windows环境下安装TVM

Table of Contents

编译环境

Visual Studio 2015 update 3以上版本、MinGW-gcc和Cmake。源码包llvm clang compiler-rt tvm。其中gcc和llvm两者版本要匹配,否则编译会出错。

  • 我的软件环境 :
  • Visual Studio 2019
  • gcc-8.1
  • Python 3.8
  • CMAKE
  • git
  • llvm 11.0
  • tvm 0.8

LLVM安装

从LLVM官网下载解压llvm、clang和compiler-rt,compiler-rt放到projects文件夹,clang放到tools中。 创建并进入build 文件夹,运行 cmake -G "Visual Studio 16 2019" .. -Thost=x64 在生成的文件中找到llvm.sln,双击打开。 选择Release x64,在右侧资源管理器右键单击解决方案,生成。 配置环境变量PATH为 llvm/build/Release/bin

TVM安装

  • 从tvm官网下载解压,修改CmakeLists.txt文件,选择需要打开的选项,其中llvm选项ON/OFF位置改为llvm-config路径,路径一般为 "llvm/build/Release/bin/llvm-config" ,使用cmake运行 cmake -G "Visual Studio 16 2019" -DCMAKE_BUILD_TYPE=Release -DCMAKE_CONFIGURATION_TYPES=Release .. 生成需要编译的文件,完成后打开tvm.sln,此部分和LLVM安装相同。编译完成后添加环境变量PYTHONPATH为 "C:\Users\hasee\tvm\python;C:\Users\hasee\tvm\build\Release;%PYTHONPATH%;"
  • 完成后根据tvm安装文档安装python包。 参考资料

https://blog.csdn.net/yunzaitian163/article/details/102629775

https://zhuanlan.zhihu.com/p/58814825

https://www.jianshu.com/p/13437251a0ab

Author: Frank

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

Validate