search

pku-liang / FlexTensor

Automatic Schedule Exploration and Optimization Framework for Tensor Computations
MIT License
174 stars 32 forks source link

Problem for importing flextensor #25

Open Arsmart1 opened 2 years ago

Arsmart1 commented 2 years ago

Hi! I am following your guide to run the tutorial code: https://pku-ahs.github.io/tutorial/en/master/steps.html#install

I have installed and pulled your docker in my WSL, and I have completed these steps:

cd FlexTensor-Micro
export PYTHONPATH=$PYTHONPATH:/path/to/FlexTensor-Micro
cd FlexTensor-Micro/flextensor/tutorial

# First, CPU experiments
cd conv2d_llvm

# run flextensor
python optimize_conv2d.py --shapes res --target llvm --parallel 8 --timeout 20 --log resnet_config.log

But I am stuck by this bug:

root@4f3f7856e4d4:/FlexTensor-Micro/flextensor/tutorial/conv2d_llvm# python3 optimize_conv2d.py --shapes res --target llvm --parallel 8 --timeout 20 --log resnet_config.log
Traceback (most recent call last):

  File "optimize_conv2d.py", line 9, in <module>
    from flextensor.utils import Config, RpcInfo

ModuleNotFoundError: No module named 'flextensor'

Could you kindly help me? Thank you!!!!

KnowingNothing commented 1 year ago

Can you show the result of

echo $PYTHONPATH

to see if flextensor is in the path?