TVM(一):简介与安装

简介

TVM是一个用于深度学习系统的编译器堆栈。

它旨在缩小以生产力为中心的深度学习框架与以性能和效率为中心的硬件后端之间的差距。TVM与深度学习框架合作,为不同的后端提供端到端编译。

换句话说,TVM就是一种将深度学习工作负载部署到硬件的端到端IR(中间表示)堆栈。可以表述为一种把深度学习模型分发到各种硬件设备上的、端到端的解决方案。

GitHub:https://github.com/dmlc/tvm

homepage:https://docs.tvm.ai/install/from_source.html

安装

1.docker镜像

1) hub网站上拉取官方提供的docker镜像

在 https://hub.docker.com/u/tvmai 进行版本查看 然后拉取 适合自己的镜像

eg: 
docker pull tvmai/demo-cpu  [此镜像中无tvm.relay]

2)官方docker脚本

官方提供了好多 dockerfile文件,可以通过官方提供的bash.sh 和 buid.sh脚本进行镜像选择以及自定义镜像制定
可通过dockerfile来查看每一个镜像里面包含的包。

eg:
git clone --recursive https://github.com/dmlc/tvm
/tvm/docker/bash.sh tvmai/demo-cpu  [tvmai/demo-cpu中也无tvm.relay]

2. 源码编译

</

评论 6
添加红包

请填写红包祝福语或标题

红包个数最小为10个

红包金额最低5元

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

抵扣说明:

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

余额充值