TVM安裝

因為現在NNVM的代碼都轉移到了TVM中，NNVM代碼也不再進行更新，因此選擇安裝的是TVM。

 

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
mkdir build
cp cmake/config.cmake build

 

 

 

修改build/config.cmake

set(USE_CUDA OFF) 到 set(USE_CUDA ON)

set(USE_LLVM OFF)  到 set(USE_LLVM ON)

cd build
cmake ..
make -j4

編譯完成

 編譯之后形成了幾個.so文件，因此我們需要修改下配置

在.bashrc文件中，添加以下

export TVM_PATH=/home/path/to/tvm
export PYTHONPATH=$TVM_PATH/python:$TVM_PATH/topi/python:$TVM_PATH/nnvm/python:${PYTHONPATH}

 這個必須要配，不然無法使用