这篇博客主要介绍环境的搭建。
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。