[BUG]: ImportError: cannot import name 'ColoInitContext' from 'colossalai.zero'

[Luoyang144]

🐛 Describe the bug

I have install colossal with CUDA_EXT=1 pip install colossalai, and get following error:
ImportError: cannot import name 'ColoInitContext' from 'colossalai.zero' (/xxx/xxxx/anaconda3/envs/colossai/lib/python3.9/site-packages/colossalai/zero/__init__.py)
Command are same as here, how can I solve this problem?

Environment

Install with requirements.txt

[MaiziXiao]

Got the same error, any idea why this happened?

[chengeharrison]

My solution is:

git clone https://github.com/hpcaitech/ColossalAI.git
cd ColossalAI
pip install .

It seems that the version pip install colossalai is installing doesn't include this change #3424 (Maybe this pr).

[Luoyang144]

@chengeharrison Thank you, it works!