因为现在NNVM的代码都转移到了TVM中,NNVM代码也不再进行更新,因此选择安装的是TVM。
git clone --recursive https://github.com/dmlc/tvmsudo apt-get updatesudo apt-get install -y python python-dev python-setuptools gcc libtinfo-dev zlib1g-devmkdir buildcp cmake/config.cmake build
修改build/config.cmake
set(USE_CUDA OFF) 到 set(USE_CUDA ON)
set(USE_LLVM OFF) 到 set(USE_LLVM ON)
cd buildcmake ..make -j4
编译完成
编译之后形成了几个.so文件,因此我们需要修改下配置
在.bashrc文件中,添加以下
export TVM_PATH=/home/path/to/tvmexport PYTHONPATH=$TVM_PATH/python:$TVM_PATH/topi/python:$TVM_PATH/nnvm/python:${PYTHONPATH}
这个必须要配,不然无法使用