TVM Windows 安装

简介

本篇博客主要目的是帮助大家在windows平台上安装好tvm，并且可以顺利使用。

因为有项目需要使用tvm，同时自己需要用windows做一些测试，因此想要在windows上安装tvm，但是安装过程中遇到了很多的挫折，反复折腾了好几次，这次终于是能用了。

同时也希望可以帮助到大家，因为我之前在网络上几乎没有看到几篇真正有用的博客。希望本篇博客可以真正起到一定的作用。

环境

系统： Windows 11

tvm版本：tvm 0.8

conda版本：4.10.3 (这个只要是最新的或者较新的其他版本应该都可以，因为官网上也推荐通过conda完成windows安装的部分工作)

工具：windows terminal (这个无所谓)

visual studio版本：vs2019 (这个官网所说2015及以上版本即可)

具体步骤

1、下载tvm(建议直接到tvm官网的下载页here进行下载，这个是比较稳定的版本，在github上下载的版本可能会有一些问题)

2、根据步骤，直接到conda这一步

# 创建环境
conda env create --file conda/build-environment.yaml
# 激活环境
conda activate tvm-build

这一步可以将大部分windows下安装所需的东西都安装好

3、进入windows编译安装的部分

# 到tvn根目录
mkdir build
# 这一步官网上没有 如果需要自己配置llvm或者cuda的话 这一步是必须的
cp cmake/config.cmake build/

cd build

# 然后根据自己需要编辑config.cmake
# set(USE_CUDA OFF) -> set(USE_CUDA ON)
# set(USE_LLVM OFF) -> set(USE_LLVM llvm_config的路径) 
# 注意llvm_config的路径一般是在conda的目录里的一个位置 也可以通过在shell里输入 which llvm_config得到 
# 例如 D:/anaconda/envs/tvm-build/Library/bin/llvm-config

cmake -A x64 -Thost=x64 ..
cd ..

cmake --build build --config Release -- /m

这一步就相当于已经完成了tvm的安装了 后续需要在python中使用tvm 还需要最后一步

4、python中进行安装

# 在tvm根目录下
cd python
python setup.py install

5、验证，在shell中输入python进入python命令行输入以下代码看是否有报错

import tvm
from tvm import te

更深层次的测试需要根据官网的教程进行校验

总结

虽然这样来看和官网的教程相比只是多了一步编辑config.cmake文件的过程，但实质上却不只如此。

因为首先官网上关于windows的安装教程非常简陋且是参杂在整个安装教程里的，很有可能走错一步就功亏一篑。

其次还需要注意conda的步骤不用走完，后面两步实质上只适合linux及相关机器安装使用，windows上安装其实是无效的。