search

hpcaitech / Open-Sora

Open-Sora: Democratizing Efficient Video Production for All
https://hpcaitech.github.io/Open-Sora/
Apache License 2.0
22.24k stars 2.18k forks source link

ModuleNotFoundError: No module named 'colossalai' #742

Open bulaikexiansheng opened 3 hours ago

bulaikexiansheng commented 3 hours ago

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? image image

bulaikexiansheng commented 2 hours ago

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.