我是靠谱客的博主 瘦瘦龙猫,最近开发中收集的这篇文章主要介绍Windows 安装 TVM + CUDA + LLVM,觉得挺不错的,现在分享给大家,希望可以做个参考。

概述

项目需要安装TVM,安装过程中踩了一些坑和解决方法:

Step 1:
git clone --recursive https://github.com/dmlc/tvm

Step 2:
​​​​ git submodule init git submodule update

注意下面标明的LLVM路径;
正如官网说的需要指明llvm-config的路径;官网写的很清楚,如果要在TVM里面用LLVM,是需要采用自己编译的方法安装LLVM的,不能用LLVM的预编译的安装方式;TVM官网并没有给出详细的原因,这是因为只有自己编译LLVM才会输出llvm-config.exe这个文件;

注意llvm-config是一个可执行文件,cmake就是通过执行他来找打llvm的其他的东西的。
用cmake编译之前就需要修改cmakefile文件,而不是之后
安装CUDA,需要注意修改好环境变量;
Windows环境变量过程可以设置一个子变量,很好用的方法:
如图:
设置“Env_extension”变量

“Env_extension”变量的值
安装CUDA的时候好像也有需要注意的地方:
一定要保证sample文件夹里面是有东西的;

研究TVM,发现下面这篇中文文章是很有用的资料:
https://iostream.io/2019/09/01/nn%e7%bc%96%e8%af%91%e6%a0%88%e4%b9%8btvm%e7%a0%94%e7%a9%b6%e6%8a%a5%e5%91%8a/

参考资料

https://docs.tvm.ai/install/from_source.html#install-contrib-libraries

最后

以上就是瘦瘦龙猫为你收集整理的Windows 安装 TVM + CUDA + LLVM的全部内容,希望文章能够帮你解决Windows 安装 TVM + CUDA + LLVM所遇到的程序开发问题。

如果觉得靠谱客网站的内容还不错,欢迎将靠谱客网站推荐给程序员好友。

本图文内容来源于网友提供,作为学习参考使用,或来自网络收集整理,版权属于原作者所有。
点赞(40)

评论列表共有 0 条评论

立即
投稿
返回
顶部