大理旅游网站建设,代运营公司排行榜,网址生成app,wordpress 优化速度关于搭建TVM编译器的官方文档#xff1a;Install from Source — tvm 0.14.dev0 documentation (apache.org)
1. 安装Anaconda
首先我们需要安装Anaconda#xff0c;因为其中包含着我们所需要的各类依赖#xff1a;
进入Anaconda官网https://www.anaconda.com/products/d…
关于搭建TVM编译器的官方文档Install from Source — tvm 0.14.dev0 documentation (apache.org)
1. 安装Anaconda
首先我们需要安装Anaconda因为其中包含着我们所需要的各类依赖
进入Anaconda官网https://www.anaconda.com/products/distribution下载Windows版本下载完成后运行.exe可以更改安装路径如果安装路径不是C盘可能需要一点点额外的步骤这个后面会提到基本一路Next就可以了。需要注意的是一定要把将Anaconda加入PATH勾选上这样就可以节省配置Anaconda环境变量的步骤。
最终安装好Anaconda可以在电脑应用菜单这里找到对应的cmd终端入口和powershell入口 2. 编译LLVM
接下来是编译LLVM简便行事我们可以使用已经手动编译好的库进入https://winlibs.com/找到包含LLVM的包 下载解压到任意目录里就行最后把其中bin目录路径写入环境变量的PATH中进入Anaconda的cmd终端中输入如下命令 llvm-config --libdir
不报错并显示对应的库目录路径意味着前面的步骤都完成了下面就开始搭建TVM。
3. 获取TVM源码
从github上获取最新版的TVM克隆到本地文件夹tvm中记住务必要克隆而非直接下载ZIP解压否则后面编译会报错 $ git clone --recursive http://github.com/apache/tvm tvm 新建一个build文件夹从文件夹cmake中复制一份config.cmake拷贝到build文件夹中 $ cd tvm$ mkdir build$ cp cmake/config.cmake build
当然源码获取并没有这么简单这也是我踩的一个大坑了如果现在查看build里的一些文件夹你会发现很多是空的这是因为它们是子仓库所以并没有跟随着一起将源码克隆到本地反而只是将一个jit链接克隆了下来我们接下来需要补充这一部分代码进入github仓库的build文件夹下将有蓝色链接含有子仓库的源码下载到本地替换原来残缺的部分 4. 构建虚拟环境
在Anaconda终端中切换到tvm的文件夹目录下注意如果你将Anaconda放在了非系统盘那么你需要使用管理员身份进入终端。
使用如下命令构建并进入tvm-build的虚拟环境 # Create a conda environment with the dependencies specified by the yamlconda env create --file conda/build-environment.yaml# Activate the created environmentconda activate tvm-build
这个过程会比较长大概10分钟左右。 5. 使用cmake生成.dll库
打开build目录下的config.cmake文件将USE_LLVM设为ON修改完后保存 进入build目录调用cmake进行生成 cd buildcmake -A x64 -Thostx64 ..cd ..cmake --build build --config Release -- /m
这里可能会有报错No CMAKE_C_COMPILER could be found意思就是没有编译器我们可以去下载Visual Studio解决这个问题如果你已经安装了Visual Studio但仍然存在这个问题那你需要去Visual Studio Installer里下载安装一个SDK这里我推荐最新版的windows 10 SDK。
这个步骤也要等待蛮久的大概半小时而且非常占用CPU电脑风扇起飞可以看到中途会有很多Warnings不必理会不是Error就知足吧Orz 最终可以看到在tvm\build\Release目录下生成了tvm.dll的链接库 6. python依赖库的安装
使用如下命令直接安装即可 pip3 install typing_extensions numpy decorator attrspip3 install tornado psutil xgboost cloudpicklepip3 install ml_dtypes 7. 将TVM安装到python环境
这一步看其他大佬的博客都是直接配置PYTHONPATH环境变量但实际上我们在Anaconda终端直接运行如下脚本即可完成 cd pythonpython setup.py install
运行完毕可以打开python的IDLE输入import tvm来简单测试一下我们搭建的TVM是否可用 如上所示没有报错基本就成功了。
当然如果不习惯用IDLE也可以使用VSCode、PyCharm等代码编辑器例如在VSCode的设置的命令面板中输入
python:select interpreter
更换python解释器为我们刚刚添加tvm包的解释器 如下所示没有报错 至此我们的TVM编译器搭建就结束了篇幅不长但是却浓缩了我踩坑的漫长经历QAQ
