search

ant-research / Pyraformer

Apache License 2.0
237 stars 37 forks source link

About the moudle of TVM #20

Open 3211345556 opened 1 year ago

3211345556 commented 1 year ago

When I set the value of use_tvm from False to True in line 270 of Single_step_main.py , Hierarchical_mm_tpm.py line 130 contains an error 'No module named 'tvm' '. When I tried to install the tvm package, it was not successfully installed. Are there any existing tvm packages available for windows system, and what should I do to solve this problem?

Zhazhan commented 1 year ago

You can compile TVM on windows as described in the TVM documentation (https://tvm.apache.org/docs/install/from_source.html#building-on-windows). Note that this repo uses TVM 0.8.0. If you feel too troubled to compile, as an alternative, you can find a compiled TVM docker image from https://tvm.apache.org/docs/install/docker.html#docker-source. Then delete files under 'pyraformer/lib' and run the code again.