一: Win10系统
0.确保电脑cuda10.2, cudnn, nvidia-smi, python3.8, vs2017,git, cmake
1.从官网下载 Pre-Built Binaries, LLVM source code, Clang source code.
2.安装预编译llvm, 安装好后命令行 clang -v验证
3.cd llvm(LLVM source code) && mkdir build && cd build
4.cmake -G "Visual Studio 15 2017 Win64" .. -Thost=x64 -DLLVM_ENABLE_PROJECTS=clang
5. 打开LLVM.sln, 选择 release x64 , 重新生成ALL_BUILD项目
6.重新生成后把Release目录下所有东西复制到预编译的LLVM的安装位置,把build里面的include和源码里面的include都复制到预编译的LLVM的安装位置的include目录
7. git clone --recursive https://github.com/apache/incubator-tvm tvm 下载源码
8.cd tvm && mkdir build && cp cmake/config.cmake build && cd build
9. 修改config.cmake 中 CUDA=ON, CUDNN=ON, LLVM=llvm-config.exe
10. cmake -G "Visual Studio 15 2017 Win64" -DCMAKE_BUILD_TYPE=Release -DCMAKE_CONFIGURATION_TYPES=Release ..
11. 打开tvm.slm, 选择 release x64, 重新生成ALL_BUILD项目
12. 安装python依赖包:pip install --user numpy decorator attrs tornado psutil xgboost , 参考官网
13. 将cl.exe(vs2017编译执行文件,基本在vs安装目录里可以找到,如果忘了可以找网上一大堆参考资料)添加进PATH环境变量
14. cd tvm && cd python && python setup.py install
15. 如果没报错,那就安装成功了,在官方教程里找一段代码跑跑看吧
二: Ubuntu系统
1. 编译安装llvm(cuda, cudnn, cmake, gcc, g++, python3.8, git这些都是要的)