Skip to content

skystoler/mlsys

Folders and files

NameName
Last commit message
Last commit date

Latest commit

 

History

23 Commits
 
 
 
 
 
 
 
 
 
 
 
 
 
 

Repository files navigation

mlsys

learn_tvm

Usage

1. Construct a virtual python env 
2. Install a nightly version tvm python3 -m  pip install mlc-ai-nightly -f https://mlc.ai/wheels

Common compilation and installtion of standard tvm

1. git clone tvm
2. cp cmake/config.cmake build
3. cmake .. && make
4. set env:
    - export TVM_HOME=/path/to/tvm
    - export PYTHONPATH=$TVM_HOME/python:$TVM_HOME/topi/python:$TVM_HOME/nnvm/python:${PYTHONPATH}

Requirements

- python >= 3.7
- torchvision
- torch
- numpy
- matplotlib
- LLVM
- CUDA
- IPython

Some commands may be helpful

1. IPython: pip3 install IPython
2. synr: pip3 install synr
3. LLVM: wget https://apt.llvm.org/llvm.sh; sudo ./llvm.sh 13

About

No description, website, or topics provided.

Resources

Stars

Watchers

Forks

Releases

No releases published

Packages

No packages published