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及以上版本即可)
conda就默认大家都安装了
具体步骤
1、下载tvm(建议直接到tvm官网的下载页here进行下载,这个是比较稳定的版本,在github上下载的版本可能会有一些问题)
2、根据步骤,直接到conda这一步
conda env create --file conda/build-environment.yaml
conda activate tvm-build
这一步可以将大部分windows下安装所需的东西都安装好
3、进入windows编译安装的部分
mkdir build
cp cmake/config.cmake build/
cd build
cmake -A x64 -Thost=x64 ..
cd ..
cmake --build build --config Release -- /m
这一步就相当于已经完成了tvm的安装了 后续需要在python中使用tvm 还需要最后一步
4、python中进行安装
cd python
python setup.py install
5、验证,在shell中输入python进入python命令行输入以下代码看是否有报错
import tvm
from tvm import te
更深层次的测试需要根据官网的教程进行校验
总结
虽然这样来看和官网的教程相比只是多了一步编辑config.cmake
文件的过程,但实质上却不只如此。
因为首先官网上关于windows的安装教程非常简陋且是参杂在整个安装教程里的,很有可能走错一步就功亏一篑。
其次还需要注意conda的步骤不用走完,后面两步实质上只适合linux及相关机器安装使用,windows上安装其实是无效的。
本文内容由网友自发贡献,版权归原作者所有,本站不承担相应法律责任。如您发现有涉嫌抄袭侵权的内容,请联系:hwhale#tublm.com(使用前将#替换为@)