官方网安装方法:从源码安裝-tvm0.8.dev0文本文档。

1.安裝Anaconda。

“连接”

2.安裝Git。

sudo apt updatesudo 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.制做后,建立一个新的转化成文件夹名称来保存文档。

mkdir buildcp cmake/config.cmake build

5.2.改动环境变量。

关键改动下列新项目:(vim能够应用/keyword突显关键词)。

假如应用CUDA开展编译程序:set(USE_CUDA ON)或是set(USE_CUDA /path/to/your/cuda/)假如应用CUDNN:set(USE_CUDNN ON)或是set(USE_CUDA /path/to/your/cudnn.so/)假如应用CUBLAS:set(USE_CUDNN ON)或是set(USE_CUDA /path/to/your/cublas.so/)假如要对IR开展debug:set(USE_RELAY_DEBUG ON)并在系统变量中添加TVM_LOG_DEBUG的途径假如要对编译程序全过程开展debug:set(USE_GRAPH_EXECUTOR ON) 和(USE_PROFILER ON)

能够依照环境变量中的详细描述改动别的作用。

5.3.编译程序。

cmake ..make -j16 (数字意思应用多少个过程开展编译程序,可依据自身的cpu参数酌情考虑挑选)

每一次改动环境变量时,都必须消除空搭建文件夹名称中除编译程序文档之外的文档并再次编译程序!

6.营造氛围。

愚昧的工程建筑自然环境:你还可以根据apt安裝llvm和其它自然环境。

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

7.安裝必需的程序包。

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

8.将TVM和CUDA加上到系统变量~/。bashrc文档。

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

评论(0条)

刀客源码 游客评论