当前位置: 首页 > 工具软件 > tvm > 使用案例 >

tvm安装与使用

窦伟
2023-12-01

主要参考:https://tvm.apache.org/docs/install/from_source.html

$git clone https://github.com/apache/incubator-tvm.git tvm && cd tvm

$git submodule update --init --recursive

$mkdir build && cd build && cp ../cmake/config.cmake .

$补齐llvm等库的路径,决定是否使用CUDA等。

$cmake ../ && make -j4

 

tvm的python前端使用:

export TVM_HOME=/path/to/tvm
export PYTHONPATH=$TVM_HOME/python:$TVM_HOME/topi/python:${PYTHONPATH}

使用教程:

 

 类似资料: