How to use different version of tvm project?

Questions
#1

Hello everyone!

I am using the tvm which is git from https://github.com/apache/tvm . Now I noticed that there is a version that can be used on nvdla backend from https://github.com/shivmgg/tvm . And I want to use the second version of TVM project.

Now my question is how can I change my tvm project to the second one. I am using anaconda to help organize the python’s environment, which means I can’t easily change to the NVDLA-TVM verison by changing my PYTHONPATH.

Hope for your answers, Thank you!

#2

If you build the TVM from source code, PYTHONPATH is always needed. Anaconda is only a package manager and is not related to the TVM version (if your TVM is not a python package installed by conda)

I’m using conda, PYTHONPTAH, and vscode to switch my TVM versions. Here is a simple example:

# ~/.zshrc
# please change the syntax a bit if you use bash
if (( !${+TVM_HOME} )); then
    export TVM_HOME=~/tvm # default tvm home
fi
export TVM_LIBRARY_PATH=${TVM_LIBRARY_PATH}:$TVM_HOME/build
export PYTHONPATH=$TVM_HOME/python:${PYTHONPATH}

# file: ${workspaceFolder}/.vscode/settings.json
# vscode settings
{
    "python.envFile": "${workspaceFolder}/.vscode/.env",
    "terminal.integrated.env.linux": {
        "TVM_HOME": "/path/to/your/version",
    },
}

# file: ${workspaceFolder}/.vscode/.env
PYTHONPATH=./python:${PYTHONPATH}