win10 安装tvm(aarch64进行推理)

发布时间:2023年12月21日

准备

  • 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 ..
    在这里插入图片描述
  2. 执行编译
    cd ../ && cmake --build build --config Release
    在这里插入图片描述

二、编译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 ..
    cmake
  4. 执行编译
    cd .. && cmake --build build --config Release -- /m
    在这里插入图片描述
  5. 查看输出E:\tvm_llvm\tvm\build\Release
    在这里插入图片描述

三、测试tvm

cd E:\tvm_llvm\tvm\python && python setup.py install
在这里插入图片描述

至此,win10平台安装完成!

文章来源:https://blog.csdn.net/wsp_1138886114/article/details/135123205
本文来自互联网用户投稿,该文观点仅代表作者本人,不代表本站立场。本站仅提供信息存储空间服务,不拥有所有权,不承担相关法律责任。