win10 安装tvm(aarch64进行推理)

文章目录

      • 准备
      • 一、编译llvm
      • 二、编译tvm
      • 三、测试tvm

准备

  • llvm下载:git clone -b v0.14.0 --depth=1 --recursive http://github.com/apache/tvm tvm
  • tvm下载:https://codeload.github.com/apache/tvm/zip/refs/tags/v0.14.0
    E:\TVM_LLVM
    ├─llvm-project-llvmorg-11.0.1.zip
    └─tvm
    
    E:根目录下创建如上tvm_llvm文件夹,并将下载的两个包放置到文件夹内。

一、编译llvm

  1. 解压llvm-project-llvmorg-11.0.1.zip 创建build并编译
    cd E:\tvm_llvm\llvm-project-llvmorg-11.0.1\llvm\ && mkdir build && cd build && cmake ..
    win10 安装tvm(aarch64进行推理)_第1张图片
  2. 执行编译
    cd ../ && cmake --build build --config Release
    win10 安装tvm(aarch64进行推理)_第2张图片

二、编译tvm

  1. 准备cmake
    cd E:\tvm_llvm\tvm && mkdir build && copy cmake\config.cmake .\build\
  2. 修改config.cmake, 若有cuda配置,可设置使用cuda
    set(USE_CUDA ON)
    set(USE_LLVM E:/tvm_llvm/llvm-project-llvmorg-11.0.1/llvm/build/Release/bin/llvm-config.exe)
  3. 切换路径执行cmake
    cd build && cmake -A x64 -Thost=x64 ..
    win10 安装tvm(aarch64进行推理)_第3张图片
  4. 执行编译
    cd .. && cmake --build build --config Release -- /m
    win10 安装tvm(aarch64进行推理)_第4张图片
  5. 查看输出E:\tvm_llvm\tvm\build\Release
    win10 安装tvm(aarch64进行推理)_第5张图片

三、测试tvm

cd E:\tvm_llvm\tvm\python && python setup.py install
win10 安装tvm(aarch64进行推理)_第6张图片

至此,win10平台安装完成!

你可能感兴趣的:(TensorRT/TVM,c++,python)