【KAWAKO】TVM-在ubuntu服务器上的安装

目录

• 下载源码
• 安装依赖库
• 修改config.cmake文件
• 编译
• 安装python库
• 添加tophub
• 简单验证

---

官方的安装教程在这里

---

下载源码

从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文件复制进去。

简单验证

开张大吉！