简介
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]