- Build libtvm.so / libtvm.dll
先安装一下llvm依赖。
https://releases.llvm.org/download.html
Windows下特别注意,要源码编译llvm,不要下载prebuilt binary。编译llvm比较简单,然后编译完了,
在tvm的build文件夹下创建一个config.cmake文件,写入这行东西:
set(USE_LLVM "D:/Program Files/llvm-project-12.0.1.src/llvm/build/Release/bin/llvm-config.exe")
将路径改为你自己的llvm路径即可。
如果下面编译提示你找不到头文件,那么就将llvm下面的include文件,拷贝到build/include里面。脑残的llvm不会给你配置这个。得手动拷贝一下。
如果你遇到zlib的问题,那么你需要从源码编译zlib,并且参照视频,vsiualstudio code里面手动更改zlib的附加项。
以下是windows例子
git clone --recursive https://github.com/apache/tvm tvm
mkdir build
cd build
# enable CUDA and llvm
cmake -A x64 -Thost=x64 -DUSE_CUDA=ON -DUSE_LLVM=ON ..
cd ..
cmake --build build --config Release -- /m
More detail at: https://tvm.apache.org/docs/install/from_source.html#python-package-installation
接着就会在:
\tvm\build\Release\tvm.dll
找到这个tvm.dll。
接着安装对应的python API:
cd python
python setup.py install