首页 > 其他 > 详细

TVM安装

时间:2018-10-01 21:59:44      阅读:249      评论:0      收藏:0      [点我收藏+]

因为现在NNVM的代码都转移到了TVM中,NNVM代码也不再进行更新,因此选择安装的是TVM

git clone --recursive https://github.com/dmlc/tvm
sudo apt-get updatesudo apt-get install -y python python-dev python-setuptools gcc libtinfo-dev zlib1g-dev
mkdir build
cp cmake/config.cmake build

修改build/config.cmake

set(USE_CUDA OFF) set(USE_CUDA ON)

set(USE_LLVM OFF)   set(USE_LLVM ON)

cd build
cmake ..
make -j4

安装完成

 

TVM安装

原文:https://www.cnblogs.com/jourluohua/p/9735841.html

(0)
(0)
   
举报
评论 一句话评论(0
关于我们 - 联系我们 - 留言反馈 - 联系我们:wmxa8@hotmail.com
© 2014 bubuko.com 版权所有
打开技术之扣,分享程序人生!