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


简单验证

 开张大吉!

  • 1
    点赞
  • 2
    收藏
    觉得还不错? 一键收藏
  • 3
    评论

“相关推荐”对你有帮助么?

  • 非常没帮助
  • 没帮助
  • 一般
  • 有帮助
  • 非常有帮助
提交
评论 3
添加红包

请填写红包祝福语或标题

红包个数最小为10个

红包金额最低5元

当前余额3.43前往充值 >
需支付:10.00
成就一亿技术人!
领取后你会自动成为博主和红包主的粉丝 规则
hope_wisdom
发出的红包
实付
使用余额支付
点击重新获取
扫码支付
钱包余额 0

抵扣说明:

1.余额是钱包充值的虚拟货币,按照1:1的比例进行支付金额的抵扣。
2.余额无法直接购买下载,可以购买VIP、付费专栏及课程。

余额充值