当前位置: 首页 > 工具软件 > tvm > 使用案例 >

tvm 入门(一)

鱼安然
2023-12-01

这篇博客主要介绍环境的搭建。

1.安装cuda和cmake,这个部分网上相关的教程很多。

2.安装llvm。llvm的版本号在4.0到6.0之间。

下载llvm源码。

链接: https://pan.baidu.com/s/1VtQeUnrj11PG6G8UGWX1gA 提取码: ikgr 。

解压该压缩文件。

tar xvJf llvm-6.0.0.src.tar.xz。

mkdir llvm-6.0.0

cd llvm-6.0.0

mkdir build

cd build

cmake -G "Unix Makefiles" -DCMAKE_BUILD_TYPE=Release  <path to llvm sources>

make -j4

sudo make install

配置环境变量

vim /etc/profile

export PATH=$PATH:llvm/build/bin(这个地方修改为之间的bin目录)

或者参考以下方式安装

wget -O - https://apt.llvm.org/llvm-snapshot.gpg.key | sudo apt-key add -
$ sudo apt-add-repository "deb http://apt.llvm.org/xenial/ llvm-toolchain-xenial-6.0 main"
$ sudo apt-get update
$ sudo apt-get install -y clang-6.0

3.安装tvm

下载tvm源码

git clone --recursive https://github.com/dmlc/tvm/
apt-get update
apt-get install -y python python-dev python-setuptools gcc libtinfo-dev zlib1g-dev

修改cmake/config.cmake文件

set(USE_CUDA ON)

set(USE_LLVM ON)

编译tvm源码

cd tvm
mkdir build
cd build
cmake ..

make -j4

设置PYTHONPATH环境变量

在安装之前先在~/.bashrc中设置PYTHONPATH环境变量

export TVM_PATH=/home/hua/Downloads/tvm/
export PYTHONPATH=$TVM_PATH/python:$TVM_PATH/topi/python:$TVM_PATH/nnvm/python:${PYTHONPATH}

对应的路径名修改为自己的。

source ~/.bashrc

参考链接:https://docs.tvm.ai/install/from_source.html#build-the-shared-library。

注:如果运行程序的时候报llvm的错误,建议重新编译llvm和tvm。

 

 

 

 

 类似资料: