当前位置：网站首页>tvm安装与使用

tvm安装与使用

2022-07-20 16:10:00 【平丘月初】

主要参考：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}

使用教程：

 

原网站

版权声明
本文为[平丘月初]所创，转载请带上原文链接，感谢
https://blog.csdn.net/u011994454/article/details/106950127

边栏推荐

猜你喜欢

随机推荐