ModuleNotFoundError: No module named 'colossalai'

[bulaikexiansheng]

I can use gradle app to finish inferencing, but when I try sequence parallel with torchrun command, this error turns out. I use pip show to check whether i install the colossalai pkg.
How can i solve this issue?

[bulaikexiansheng]

It seems that the path of pytorch that I installed using pip is not in the sora virtual environment, which may be the reason for the error. I am re-downloading pytorch using conda.