TVM安装

Posted jourluohua

tags:

篇首语:本文由小常识网(cha138.com)小编为大家整理,主要介绍了TVM安装相关的知识,希望对你有一定的参考价值。

因为现在NNVM的代码都转移到了TVM中,NNVM代码也不再进行更新,因此选择安装的是TVM

git clone --recursive https://github.com/dmlc/tvm
sudo apt-get updatesudo 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

安装完成

 

以上是关于TVM安装的主要内容,如果未能解决你的问题,请参考以下文章

TVM实战

TVM实战

TVM实战

TVM实战

tvm在linux环境下的安装与编译及vscode如何配置tvm的远程连接调试环境

tvm在linux环境下的安装与编译及vscode如何配置tvm的远程连接调试环境