最近想想学习一下TVM,结果只是安装编译就花了一天多的时间。由于环境不同,官网的安装教程有很多坑需要自己趟,特此记录一下,希望能帮大家节省时间。
一. Ubuntu20.04安装配置
由于要用到RPC来进行auto-tuning,所以Ubuntu20.04的网络适配器需要选择桥接模式,VMware也需要编辑虚拟网络适配器,添加桥接模式。Win10的WSL2默认的是NAT模式,最新的WSL2可以支持桥接,但是需要升级系统比较麻烦,本文中采用Ubuntu20.04。
注:安装VMware及Ubuntu20.04,建议虚拟盘内存设置大一些,我的环境中设置的是100G。
二. 安装编译工具
2.1 安装git/vim/gcc/g++/make
sudo apt-get install git vim gcc g++ make
2.2 安装cmake
cmake最好由源码安装,apt-get安装的版本比较老,本环境安装:cmake-3.25.1.tar.gz
sudo apt-get install libssl-dev #不安装编译会报错: Could not find OpenSSL
./bootstrap
make -j4
sudo make install
运行cmake --version显示版本号则安装正确。
2.3 安装LLVM,本环境安装:clang+llvm-13.0.0
xz -d clang+llvm-13.0.0-x86_64-linux-gnu-ubuntu-20.04.tar.xz
tar -xvf clang+llvm-13.0.0-x86_64-linux-gnu-ubuntu-20.04.tar
mv clang+llvm-13.0.0-x86_64-linux-gnu-ubuntu-20.04 clang+llvm-13.0.0
添加llvm环境变量,后source ~/.bashrc
echo 'export LLVM_HOME=/opt/clang+llvm-13.0.0/bin' >> ~/.bashrc
echo 'export PATH=$LLVM_