Ubuntu安装TVM

TVM安装

0.环境最低要求

• 支持C++11的C++编译器(4.8版本以上的gcc)
• cmake >= 3.5
• 如果需要，配置好CUDA和LLVM

1.下载TVM源码

在你想要放置tvm的地方，执行如下命令，会新建一个tvm文件夹，并将源码下载到其中

git clone --recursive https://github.com/apache/incubator-tvm tvm

2.进入tvm文件夹，复制cmake配置文件到创建的build目录中，根据自己的需求进行配置

cd tvm
mkdir build
cp cmake/config.cmake build
vi config.cmake

几个常用的设置

• 使用CUDA：

  找到

  set(USE_CUDA OFF)
  

  改为

  set(USE_CUDA ON)
  

  会自动寻找当前环境中的CUDA。

• 使用LLVM

  找到

  set(USE_LLVM OFF)
  

  改为

  set(USE_LLVM 你的llvm-config位置)
  

  如我的为

  set(USE_LLVM /usr/lib/llvm-8/bin/llvm-config)
  

  官方文档中说可以直接设为ON，让cmake自动寻找，因为我的环境中有好几个版本的llvm，因此自己选择手动设置。

3.编译

cd build
cmake ..
make -j4

如果到达100%，并且没有报错，说明源码编译好了。

4.Python接口配置

进入自己的.bashrc文件

sudo vi ~/.bashrc

在文件的末尾添加以下内容：

export TVM_HOME=你的TVM安装路径
export PYTHONPATH=$TVM_HOME/python:$TVM_HOME/topi/python:${PYTHONPATH}

安装python依赖：

pip3 install --user numpy decorator attrs tornado psutil xgboost antlr4

前三个是必须的，中间三个是auto-turning时要用到的，最后一个是编译器的语法分析器。

5.安装googletest进行c++测试

在想要安装googletest的位置，依次执行如下命令：

git clone https://github.com/google/googletest
cd googletest
mkdir build
cd build
cmake ..
make
make install

安装完成后，回到TVM的安装目录，执行如下命令：

./tests/scripts/task_cpp_unittest.sh

可以看到TVM的C++ 测试情况，如果可以看到全为passed，则说明安装成功。