官方的安装教程在这里


下载源码

从Download Page中下载下来的源码是0.8版本的,亲测该版本不可用,需要从git中clone最新版(当前最新版为0.9dev版)。

命令中的“--recursive”为“递归下载”的意思,因为TVM仓库中包含了别的仓库的代码,需要全部下载下来。

在git clone的时候可能会遇到死活连接不上因此无法开始下载的问题,如下图。网上说可以配置git clone代理来解决,但试了几次都不太行。第二天来公司不信邪又clone一下,突然又下下来了。。。这个就还是看运气吧。

在clone过程中可能会存在一些第三方库(在“./3rdparty”中)下不下来,这个可以自行去下载然后添加到工程中。

补充:我将包含第三方库的TVM0.9.dev0版本打包传到了csdn上,有需要的可以自取。

安装依赖库

按照官方手册中的命令将需要的依赖库安装上。

修改config.cmake文件

在工程根目录下新建一个build文件夹,将cmake/config.cmake文件复制进去。可以根据自身需求对其中一些选项进行修改,如USE_CUDA等。(我按照官方文档将USE_RELAY_DEBUG设置为了ON)

然后从官方文档中的LLVM Download Page中寻找适合自己系统的预编译好的LLVM进行下载,比如我的是ubuntu18就选择这个。

然后根据官方手册修改config.cmake中的“set(USE_LLVM OFF)”,将OFF替换为刚才下载下来的LLVM中bin文件夹中的llvm-config

编译

老三连

安装python库

使用第二种方法的这条命令,不需要运行上面的那条export(mac系统才用)。

添加tophub

在“~/.tvm”路径下有个tophub文件夹,若前面下载成功,则里面会有一大堆.log文件,若未成功则文件夹为空。

若文件夹为空,我们需要到tophub的github仓库中手动下载,然后将这些.log文件复制进去。

简单验证

开张大吉!

posted on 2023-01-31 13:22  KAWAKO  阅读(119)  评论(0编辑  收藏  举报