ICode9

精准搜索请尝试: 精确搜索
首页 > 系统相关> 文章详细

Ubuntu下TVM的编译安装

2019-06-30 18:26:06  阅读:574  来源: 互联网

标签:LLVM apt TVM 编译 llvm install Ubuntu 安装


官方安装文档地址: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

标签:LLVM,apt,TVM,编译,llvm,install,Ubuntu,安装
来源: https://blog.csdn.net/weixin_43953703/article/details/94344889

本站声明: 1. iCode9 技术分享网(下文简称本站)提供的所有内容,仅供技术学习、探讨和分享;
2. 关于本站的所有留言、评论、转载及引用,纯属内容发起人的个人观点,与本站观点和立场无关;
3. 关于本站的所有言论和文字,纯属内容发起人的个人观点,与本站观点和立场无关;
4. 本站文章均是网友提供,不完全保证技术分享内容的完整性、准确性、时效性、风险性和版权归属;如您发现该文章侵犯了您的权益,可联系我们第一时间进行删除;
5. 本站为非盈利性的个人网站,所有内容不会用来进行牟利,也不会利用任何形式的广告来间接获益,纯粹是为了广大技术爱好者提供技术内容和技术思想的分享性交流网站。

专注分享技术,共同学习,共同进步。侵权联系[81616952@qq.com]

Copyright (C)ICode9.com, All Rights Reserved.

ICode9版权所有