【环境搭建】ubuntu18.04 编译tvm

最新推荐文章于 2024-10-10 17:35:29 发布

原创最新推荐文章于 2024-10-10 17:35:29 发布 · 405 阅读

1 ·

CC 4.0 BY-SA版权

环境搭建专栏收录该内容

74 篇文章

订阅专栏

本文档详细介绍了如何编译和安装TVM项目，包括使用apt-get安装依赖，从GitHub克隆源码，配置编译选项，以及生成Python环境所需的库文件。在编译过程中，特别指出了开启GRAPH_RUNTIME_DEBUG选项，并在.bashrc中设置TVM环境变量，以确保Python环境的正确配置。

部署运行你感兴趣的模型镜像

2，编译和安装
2.1 编译

sudo apt-get install clang-6.0 lldb-6.0
git clone https://github.com/apache/tvm.git
cd tvm
git submodule sync
git submodule update --init --recursive
mkdir build
cp cmake/config.cmake build
cd build
vim config.cmake 
GRAPH_RUNTIME_DEBUG = ON
# 指令
cmake ..  # cmake版本 > 3.17.0
make -j 8

2.2 配置
tvm项目的build会生成如下2个文件

libtvm_runtime.so
libtvm.so

2.3 tvm -> python环境

vim ~/.bashrc
export TVM_HOME=/path/to/tvm
export PYTHONPATH=$TVM_HOME/python:$TVM_HOME/topi/python:$TVM_HOME/nnvm/python:${PYTHONPATH}
source ~/.bashrc

您可能感兴趣的与本文相关的镜像