win10 安装tvm（aarch64进行推理）

2023-12-21 22:03:45

文章目录

准备

  • 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
本文来自互联网用户投稿，该文观点仅代表作者本人，不代表本站立场。本站仅提供信息存储空间服务，不拥有所有权，不承担相关法律责任。