Ubuntu20.04上编译安装TVM

本文主要讲述如何在ubuntu20.04平台上编译TVM代码并在python中import tvm成功。

源代码下载:

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

平台环境升级:

1) sudo apt-get update

2) sudo apt-get install -y python3 python3-dev python3-setuptools gcc libtinfo-dev zlib1g-dev build-essential cmake libedit-dev libxml2-dev

cmake时遇到cmake版本过低的问题(当前是3.16,但tvm源码的CMakeLists.txt要求至少3.18):

cd tvm/

sudo mkdir build

cd build/

sudo chmod 777 -R ./

cmake ..

想通过apt-get install cmake==xxx来安装指定版本的cmake但失败:

aptitude versions cmake

sudo apt-get update

sudo apt-get install aptitude

aptitude versions cmake #寻找可以安装的cmake版本

sudo apt-get install cmake=3.20.2 #失败

sudo apt-get install cmake=3.20.0 #失败

通过源代码来编译安装cmake指定版本成功!

cd /work/

wget https://cmake.org/files/v3.22/cmake-3.22.0.tar.gz

tar xvzf cmake-3.22.0.tar.gz

cd cmake-3.22.0/

./configure

./configure时遇到下面的错误:

CMake Error at Utilities/cmcurl/CMakeLists.txt:562 (message): Could not find OpenSSL. Install an OpenSSL development package or configure CMake with -DCMAKE_USE_OPENSSL=OFF to build without OpenSSL.

通过下面命令解决并./configure成功

sudo apt-get install libssl-dev

./configure

cmake安装成功,并显示版本号为3.22

sudo make install

cmake --version

解决cmake版本问题后,tvm代码就顺利编译成功

cd tvm/build/

cp ../cmake/config.cmake build

cmake ..

make -j16

选择适当的LLVM已经编译好的安装包并下载

sudo wget https://github.com/llvm/llvm-project/releases/download/llvmorg-13.0.0/clang+llvm-13.0.0-x86_64-linux-gnu-ubuntu-20.04.tar.xz

tar -xf clang+llvm-13.0.0-x86_64-linux-gnu-ubuntu-20.04.tar.xz

在tvm中指定llvm binary所在的路径

vi /work/tvm/build/config.cmake,并作如下修改

重新编译tvm

cd /work/tvm/build

make clean;make -j16

最后在python环境验证tvm是否安装成功:

需要先安装pip,numpy以及decorator安装包

sudo apt install python3-pip

pip3 install numpy

pip3 install decorator

并将tvm的安装路径添加到PYTHONPATH

export TVM_HOME=/work/tvm

export PYTHONPATH=TVM_HOME/python:PYTHONPATH

最后python中成功import tvm:

相关推荐
周之鸥4 小时前
Qt 项目国际化从零到一:用 Qt Linguist 实现多语言动态切换(含源码与踩坑指南)
qt·i18n·cmake·qmake·linguist·lupdate·lrelease
沢田纲吉1 天前
《LLVM IR 学习手记(五):关系运算与循环语句的实现与解析》
前端·c++·llvm
沢田纲吉1 天前
《LLVM IR 学习手记(六):break 语句与 continue 语句的实现与解析》
前端·c++·llvm
老黄编程3 天前
ros2 中 CMakeLists.txt 的 ament_package 有什么用?有什么使用约束?必须放置尾部吗?
ros·cmake
澄澈i3 天前
CMake学习篇[3]---CMake进阶+嵌套CMakeLists+多层级关系
c++·学习·cmake
一只小bit9 天前
CMake 入门实战手册:从理解原理开始,打造高效 C/C++ 开发流程
c语言·开发语言·c++·cmake
沢田纲吉10 天前
《LLVM IR 学习手记(三):赋值表达式与错误处理的实现与解析》
前端·编程语言·llvm
研來如此14 天前
Visual Studio 2022
cmake·visual studio
沢田纲吉17 天前
《LLVM IR 学习手记(二):变量表达式编译器的实现与深入解析》
前端·编程语言·llvm
HyperAI超神经18 天前
【TVM 教程】设置 RPC 系统
开发语言·网络·人工智能·python·网络协议·rpc·tvm