2207-TVM在Ubuntu系统下的安装

在Ubuntu系统下通过conda环境安装tvm

安装过程主要参看官方文档，不过那个文档考虑的方面太多了，容易绕晕。这里考虑在conda环境下安装tvm。

0.服务器环境

系统：Linux version 5.4.0-107-generic (buildd@lcy02-amd64-058) (gcc version 9.4.0 (Ubuntu 9.4.0-1ubuntu1~20.04.1))
cpu：Intel(R) Xeon(R) CPU E5-2695 v4 @ 2.10GHz

1.从GitHub获取源码

git clone --recursive https://github.com/apache/tvm tvm。利用--recursive选项把子模块也进行克隆

2.构建Conda环境

在./tvm/conda/文件夹下有一个build-environment.yaml文件，利用它可以直接构建出安装tvm需要的前置环境。

# Create a conda environment with the dependencies specified by the yaml
conda env create --file conda/build-environment.yaml
# Activate the created environment
conda activate tvm-build

注意python版本,3.9.X以上版本的python是不被支持的。
接着就是在conda环境中安装tvm了。

conda build --output-folder=conda/pkg  conda/recipe
# Run conda/build_cuda.sh to build with cuda enabled
conda install tvm -c ./conda/pkg

正常进行到这里的话，tvm-build环境中已经有tvm了，可以进行测试

3.总结

利用conda环境来安装tvm还是相对容易的，毕竟conda的包管理确实好，而且tvm也支持直接利用conda来安装。