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的其他的东西的。
Windows 安装 TVM + CUDA + LLVM_第1张图片
安装CUDA,需要注意修改好环境变量;
Windows环境变量过程可以设置一个子变量,很好用的方法:
如图:
Windows 安装 TVM + CUDA + LLVM_第2张图片

Windows 安装 TVM + CUDA + LLVM_第3张图片
安装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

你可能感兴趣的:(TVM,TVM,CUDA,LLVM,Windows)