Ubuntu下TVM的编译安装

内容导读

互联网集市收集整理的这篇技术教程文章主要介绍了Ubuntu下TVM的编译安装，小编现在分享给大家，供广大互联网技能从业者学习和参考。文章包含1856字，纯文字阅读大概需要3分钟。

内容图文

官方安装文档地址：https://docs.tvm.ai/install/index.html
系统环境：Ubuntu 18.04 LTS 64-bit，其他环境或需求可参考官方文档。

TVM编译安装

LLVM部分

虽然LLVM对于TVM是可选项，但是如果要部署到CPU端，那么LLVM几乎是必须的，所以建议安装LLVM。本次安装LLVM6.0。
第一步，添加相关源
编辑/etc/apt/sources.list，将以下源加入：

deb http://apt.llvm.org/xenial/ llvm-toolchain-xenial-6.0 main
deb-src http://apt.llvm.org/xenial/ llvm-toolchain-xenial-6.0 main     

加入完成后需取得数字证书

wget -O - http://llvm.org/apt/llvm-snapshot.gpg.key|sudo apt-key add -

然后务必进行apt-get update
第二步，安装LLVM

apt-get install clang-6.0 lldb-6.0

其他操作系统和LLVM版本参考：https://apt.llvm.org/

TVM部分

第一步，从GitHub克隆TVM的repo

git clone --recursive https://github.com/dmlc/tvm

注意此处的--recursive是必须的，否则在编译时会出错。
第二步，安装前准备

sudo apt-get update
sudo apt-get install -y python python-dev python-setuptools gcc libtinfo-dev zlib1g-dev build-essential cmake

第三步，修改config.cmake配置
建立build文件夹，复制config.cmake。

mkdir build
cp cmake/config.cmake build

然后修改LLVM配置，将set(USE_LLVM OFF)改为set(USE_LLVM ON)
第四步，启动编译

cd build
cmake ..
make -j4

如果顺利完成，即可进入Python包安装

Python包安装

此处使用官方文档推荐的第一种方法进行Python包安装。这种方法可以使我们更改源代码并重新编译后，无需对Python端进行任何更改。
第一步，编辑~/.bashrc，在文件末尾添加：

export TVM_HOME=/path/to/tvm
export PYTHONPATH=$TVM_HOME/python:$TVM_HOME/topi/python:$TVM_HOME/nnvm/python:${PYTHONPATH}

其中/path/to/tvm为tvm目录。
第二步，使修改立即生效

source ~/.bashrc

最后安装其他Python依赖
必须的依赖：

pip install --user numpy decorator attrs

如果需要RPC Tracker：

pip install --user tornado

如果需要auto-tuning模块：

pip install --user tornado psutil xgboost

解析Relay text format程序必须使用Python 3并安装：

pip install --user mypy orderedset antlr4-python3-runtime

内容总结

以上是互联网集市为您收集整理的Ubuntu下TVM的编译安装全部内容，希望文章能够帮你解决Ubuntu下TVM的编译安装所遇到的程序开发问题。 如果觉得互联网集市技术教程内容还不错，欢迎将互联网集市网站推荐给程序员好友。

内容备注

版权声明：本文内容由互联网用户自发贡献，该文观点与技术仅代表作者本人。本站仅提供信息存储空间服务，不拥有所有权，不承担相关法律责任。如发现本站有涉嫌侵权/违法违规的内容， 请发送邮件至 gblab@vip.qq.com 举报，一经查实，本站将立刻删除。

内容手机端

---