官方安裝方法：Install from Source — tvm 0.8.dev0 documentation

1. 安裝Anaconda

「鏈接」

2. 安裝Git

sudo apt update
sudo apt install git

3. 從GitHub上下載tvm源代碼

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

4. 安裝必要的依賴

sudo apt install -y gcc libtinfo-dev zlib1g-dev build-essential cmake libedit-dev libxml2-dev

5. 修改TVM編譯配置文件

cd tvm

如果沒有vim，請先安裝vim，再對配置文件進行修改

sudo apt install vim 

5.1、新建一個build文件夾用於存放make後的文件

mkdir build
cp cmake/config.cmake build

5.2、修改配置文件

主要修改如下幾項：（vim使用/關鍵字可以高亮關鍵字）

1. 如果使用CUDA進行編譯：set(USE_CUDA ON)或者set(USE_CUDA /path/to/your/cuda/)
2. 如果使用CUDNN：set(USE_CUDNN ON)或者set(USE_CUDA /path/to/your/cudnn.so/)
3. 如果使用CUBLAS：set(USE_CUDNN ON)或者set(USE_CUDA /path/to/your/cublas.so/)
4. 如果要對IR進行debug：set(USE_RELAY_DEBUG ON)並在環境變量中加入TVM_LOG_DEBUG的路徑
5. 如果要對編譯過程進行debug：set(USE_GRAPH_EXECUTOR ON) 和(USE_PROFILER ON)

其餘功能可按照配置文件中的詳細說明進行修改

5.3、編譯

cmake ..
make -j16  (數字代表使用多少個進程進行編譯，可根據自己的cpu參數酌情選擇)

每一次修改配置文件之後都需要把build文件夾中的除編譯文件外的文件清空，重新編譯！

6. 搭建環境

傻瓜式搭建環境：也可以自行通過apt安裝llvm等環境

conda env create --file conda/build-environment.yaml
conda activate tvm-build

7. 安裝必要的包

conda build --output-folder=conda/pkg  conda/recipe
conda install tvm -c ./conda/pkg
pip install numpy decorator attrs tornado psutil xgboost cloudpickle

8. 把TVM和CUDA加入到環境變量~/.bashrc文件中

export TVM_HOME=/path/to/tvm
export PYTHONPATH=$TVM_HOME/python:${PYTHONPATH}
export PATH=/path/to/cuda:$PATH
export LD_LIBRARY_PATH=/path/to/cuda/lib64:$LD_LIBRARY_PATH