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

 

git clone --recursive https://github.com/dmlc/tvm
sudo apt-get update
sudo 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

编译完成

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

在.bashrc文件中,添加以下

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

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

相关文章:

  • 2021-07-12
  • 2022-01-11
  • 2021-08-14
  • 2021-11-17
  • 2021-07-20
  • 2022-02-15
  • 2021-06-17
  • 2022-12-23
猜你喜欢
  • 2021-06-23
  • 2021-08-30
  • 2021-09-04
  • 2022-02-03
  • 2021-12-13
  • 2021-09-17
  • 2021-06-14
相关资源
相似解决方案