Linux下安装TVM

TVM安装

前言

安装要求：

• gcc版本 >=4.8
• CMake >=3.5
• python3 最新的tvm已经不支持python2了
• llvm 我选择的版本为llvm-4.0.0

从github上下载TVM

下载源码->创建要编译生成so的文件夹->

	$ git clone --recursive https://github.com/dmlc/tvm
	$ sudo apt-get update
	$ sudo apt-get install -y python python-dev python-setuptools gcc libtinfo-dev zlib1g-dev build-essential cmake

创建build文件，并拷贝修改config配置文件

在tvm目录下创建build文件，并将tvm/cmake下的config.camke文件拷贝到build目录下

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

打开build下的config.cmake文件，因为我需要支持CUDA与llvm环境，所以找到下面的配置并设置ON

set(USE_CUDA OFF)     --->set(USE_CUDA ON)  
set(USE_LLVM OFF)     --->set(USE_LLVM ON)

如果要使用cudnn，就去打开cudnn的开关，根据自己的需求来，还有一个开关如果你需要对源码进行调试跟踪建议打开

set(USE_GRAPH_RUNTIME_DEBUG ON)

编译

修改好配置文件后，进行编译。因为修改了两个编译选项，因此首先需要cmake重新生成Makefile，以后每次新添加了文件和文件夹，一定要重新cmake，否则文件很可能没有编译。

  	$ cd build
  	$ cmake -DCMAKE_BUILD_TYPE=Debug ..   //如果需要gdb跟踪源码的话需要加-DCMAKE_BUILD_TYPE=Debug
   	$ make -j4

添加环境变量

$ vim ~/.bashrc
添加：
export TVM_PATH=yourpath/to/tvm （你自己的目录）
export PYTHONPATH=$TVM_PATH/python:$TVM_PATH/topi/python:$TVM_PATH/nnvm/python:${PYTHONPATH}
source ~/.bashrc

官网安装教程：

https://docs.tvm.ai/install/from_source.html