This page gives instructions on how to build and install the nnvm compiler package from scratch on various systems. It consists of two steps:
- First build the shared library from the C++ codes (
libnnvm_compiler.sofor linux/osx and
- Setup for the language packages (e.g. Python Package).
To get started, clone nnvm repo from github. It is important to clone the submodules along, with
git clone --recursive https://github.com/dmlc/nnvm
For windows users who use github tools, you can open the git shell, and type the following command.
git submodule init git submodule update --recursive
NNVM compiler depend on TVM and TOPI, so make sure you install them by following TVM document. Note that it is necessary to build TVM with LLVM support to take full benefit of NNVM compiler.
Python Package Installation¶
The python package is located at python. There are several ways to install the package, in all these cases the TVM library must be present in the python env:
Set the environment variable
PYTHONPATHto tell python where to find the library. For example, assume we cloned
nnvmon the home directory
~. then we can added the following line in
~/.bashrc. It is recommended for developers who may change the codes. The changes will be immediately reflected once you pulled the code and rebuild the project (no need to call
Install nnvm python bindings by
# install nnvm package for the current user # NOTE: if you installed python via homebrew, --user is not needed during installaiton # it will be automatically installed to your user directory. # providing --user flag may trigger error during installation in such case. cd python; python setup.py install --user; cd ..