[BUG]: from colossalai._C import fused_optim ImportError

[yufengyao-lingoace]

🐛 Describe the bug

/examples/images/dreambooth/ runs error as follows:

Using /home/ubuntu/.cache/torch_extensions/py37_cu117 as PyTorch extensions root... Traceback (most recent call last): File "/home/ubuntu/anaconda3/lib/python3.7/site-packages/colossalai/kernel/init.py", line 4, in from colossalai._C import fused_optim ImportError: /home/ubuntu/anaconda3/lib/python3.7/site-packages/colossalai/_C/fused_optim.cpython-37m-x86_64-linux-gnu.so: undefined symbol: _ZN2at4_ops19empty_memory_format4callEN3c108ArrayRefIlEENS2_8optionalINS2_10ScalarTypeEEENS5_INS2_6LayoutEEENS5_INS2_6DeviceEEENS5_IbEENS5_INS2_12MemoryFormatEEE

Environment

No response

[feifeibear]

Did you install it from our official website. If install from source, please check if your CUDA tookit version matches your torch and colossalai version.

[yufengyao-lingoace]

Yes, I just folows your official document:

[Issues-translate-bot]

Bot detected the issue body's language is not English, translate it automatically. 👯👭🏻🧑‍🤝‍🧑👫🧑🏿‍🤝‍🧑🏻👩🏾‍🤝‍👨🏿👬🏿

---

Yes, I just folows your official document:

[yufengyao-lingoace]

my cuda is 11.4

[haofanwang]

You can try with cuda 11.3 or 11.6, and install the corresponding Pytorch.

[weiyuann1]

my cuda is 11.4

I met the same question. Have you salved it by changing cuda version?

[gouchangjiang]

you may solve this problem if you use a higher version of gcc. I solved the same sort of problem using gcc 7.3.0.

[binmakeswell]

you may solve this problem if you use a higher version of gcc. I solved the same sort of problem using gcc 7.3.0.

Thanks for the advice. We have updated a lot. This issue was closed due to inactivity. If you have similar bugs, please open a new issue. Thanks.