TVM:深度学习框架编译器的安装踩坑集


一、引言

4月底的时候,花了一天时间安装TVM这个东西。那时,上来就弄TVM 0.8的版本,但因为具体安装的流程不是很懂,所以出现了一些莫名其妙的问题。后来,直接改用TVM0.6版本,一路顺风。

但,今天(5月7日)时,打算仔细看下教程,发现0.6版本的,已经没了,只剩下0.8的教程,行吧!还是得回头安装0.8的,但之前怎么安装的,又给忘了,于是乎,一顿操作后,直接弄好了,打算写篇文章讲讲这个东西的安装。


二、流程

安装的流程很简单,网上也有许多教程了,这里先详细地讲下。

1°代码的获取。

使用“git clone --recursive + 网址”的方式,可以完成代码的获取。因为在tvm开源包的3rdparty文件夹中存在dlpack、dmlc-core、rang和vta-hw等库,如果不加 --recursive 选项,会缺失掉。

在我的实际测试中发现,如果使用git clone来获取源码,网速问题还是GitHub本身原因,没到一半就GG掉了,几次下来都没搞成功。也许需要解决网速上的问题,但得不偿失,用下面这个方法处理:

打开下面网址。选择版本,点击↓Code下载代码包。

https://github.com/apache/tvm

在这里插入图片描述
打开3rdparty文件夹,点击标蓝的几个库,使用↓Code把对应的文件全都下载了。
在这里插入图片描述
手动解压后,把各文件放在对应的目录下。这里我把整合后的文件打包了,放在自己的GitHub上,方便下次使用。包含0.8和0.6的版本。
在这里插入图片描述
附上打包文件的地址:https://github.com/abeitian/tvm_package

2°安装Linux的依赖。

这里使用的是WSL的Linux Ubuntu18.04、GCC7.5和Python3.7.3环境,网上有关于Windows安装的教程,但需要VS2019,我的电脑已经快承受不了了,所以在0.6没成功后,还是选择了硬刚。

安装依赖命令。这个一般不会错,如果提示少了,就再多装下。

sudo apt-get update
sudo apt-get inst
评论 5
添加红包

请填写红包祝福语或标题

红包个数最小为10个

红包金额最低5元

当前余额3.43前往充值 >
需支付:10.00
成就一亿技术人!
领取后你会自动成为博主和红包主的粉丝 规则
hope_wisdom
发出的红包

打赏作者

ATian+

你的鼓励将是我创作的最大动力

¥1 ¥2 ¥4 ¥6 ¥10 ¥20
扫码支付:¥1
获取中
扫码支付

您的余额不足,请更换扫码支付或充值

打赏作者

实付
使用余额支付
点击重新获取
扫码支付
钱包余额 0

抵扣说明:

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

余额充值