win10 安装tvm(aarch64进行推理)
准备
- tvm下载:
git clone -b v0.14.0 --depth=1 --recursive http://github.com/apache/tvm tvm - llvm下载:https://codeload.github.com/llvm/llvm-project/zip/refs/tags/llvmorg-11.0.1
在E:\TVM_LLVM ├─llvm-project-llvmorg-11.0.1.zip └─tvmE:根目录下创建如上tvm_llvm文件夹,并将下载的两个包放置到文件夹内。
一、编译llvm
- 解压llvm-project-llvmorg-11.0.1.zip 创建build并编译
cd E:\tvm_llvm\llvm-project-llvmorg-11.0.1\llvm\ && mkdir build && cd build && cmake ..

- 执行编译
cd ../ && cmake --build build --config Release

二、编译tvm
- 准备cmake
cd E:\tvm_llvm\tvm && mkdir build && copy cmake\config.cmake .\build\ - 修改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) - 切换路径执行cmake
cd build && cmake -A x64 -Thost=x64 ..

- 执行编译
cd .. && cmake --build build --config Release -- /m

- 查看输出
E:\tvm_llvm\tvm\build\Release

三、测试tvm
cd E:\tvm_llvm\tvm\python && python setup.py install

至此,win10平台安装完成!