1. Build libtvm.so / libtvm.dll

    先安装一下llvm依赖。

    https://releases.llvm.org/download.html

    Windows下特别注意,要源码编译llvm,不要下载prebuilt binary。编译llvm比较简单,然后编译完了,
    在tvm的build文件夹下创建一个config.cmake文件,写入这行东西:

    1. 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例子

    1. git clone --recursive https://github.com/apache/tvm tvm
    2. mkdir build
    3. cd build
    4. # enable CUDA and llvm
    5. cmake -A x64 -Thost=x64 -DUSE_CUDA=ON -DUSE_LLVM=ON ..
    6. cd ..
    7. cmake --build build --config Release -- /m

    More detail at: https://tvm.apache.org/docs/install/from_source.html#python-package-installation
    接着就会在:

    1. \tvm\build\Release\tvm.dll

    找到这个tvm.dll。

    接着安装对应的python API:

    1. cd python
    2. python setup.py install