博客
关于我
强烈建议你试试无所不能的chatGPT,快点击我
TVM安装
阅读量:5079 次
发布时间:2019-06-12

本文共 604 字,大约阅读时间需要 2 分钟。

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

 

git clone --recursive https://github.com/dmlc/tvmsudo apt-get updatesudo apt-get install -y python python-dev python-setuptools gcc libtinfo-dev zlib1g-devmkdir buildcp cmake/config.cmake build

 

 

 

修改build/config.cmake

set(USE_CUDA OFF) set(USE_CUDA ON)

set(USE_LLVM OFF)   set(USE_LLVM ON)

cd buildcmake ..make -j4

编译完成

 编译之后形成了几个.so文件,因此我们需要修改下配置

在.bashrc文件中,添加以下

export TVM_PATH=/home/path/to/tvmexport PYTHONPATH=$TVM_PATH/python:$TVM_PATH/topi/python:$TVM_PATH/nnvm/python:${PYTHONPATH}

 这个必须要配,不然无法使用

转载于:https://www.cnblogs.com/jourluohua/p/9735841.html

你可能感兴趣的文章
solr后台操作Documents之增删改查
查看>>
http://yusi123.com/
查看>>
文件文本的操作
查看>>
Ubuntu linux下gcc版本切换
查看>>
记一次Web服务的性能调优
查看>>
jQuery.form.js使用
查看>>
(转)linux sort,uniq,cut,wc命令详解
查看>>
关于ExecuteNonQuery执行的返回值(SQL语句、存储过程)
查看>>
UVa540 Team Queue(队列queue)
查看>>
mysql数据增删改查
查看>>
shell中下载最新版本或指定版本的办法(Dockerfile 中通用)
查看>>
极客时间-左耳听风-程序员攻略-分布式架构工程设计
查看>>
akka之种子节点
查看>>
不知道做什么时
查看>>
matlab 给某一列乘上一个系数
查看>>
密码学笔记——培根密码
查看>>
Screening technology proved cost effective deal
查看>>
MAC 上升级python为最新版本
查看>>
创业老板不能犯的十种错误
查看>>
Animations介绍及实例
查看>>