[BUG]: pip install failed when CUDA_EXT is 1

[fuhengwu2021]

🐛 Describe the bug

$CUDA_EXT=1 python -m pip install --force-reinstall --no-deps colossalai -U
Looking in indexes: https://pypi.org/simple, https://pypi.ngc.nvidia.com
Collecting colossalai
  Downloading colossalai-0.2.0.tar.gz (697 kB)
     ━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━ 697.8/697.8 kB 10.7 MB/s eta 0:00:00
  Preparing metadata (setup.py) ... done
Discarding https://files.pythonhosted.org/packages/9c/c7/427dd2ed3769c6beb05cefdc18f8fad634673ed1472d5409afbdb768c7c3/colossalai-0.2.0.tar.gz (from https://pypi.org/simple/colossalai/) (requires-python:>=3.6): Requested colossalai from https://files.pythonhosted.org/packages/9c/c7/427dd2ed3769c6beb05cefdc18f8fad634673ed1472d5409afbdb768c7c3/colossalai-0.2.0.tar.gz has inconsistent version: expected '0.2.0', but metadata has '0.2.0+torch1.13cu11.3'
ERROR: Could not find a version that satisfies the requirement colossalai (from versions: 0.2.0)
ERROR: No matching distribution found for colossalai

Environment

linux ubuntu

[FrankLeeeee]

Hi @fuhengwu2021 , I am able to reproduce this problem. Thank you for discovering this bug. I will fix it tomorrow.

[FrankLeeeee]

Hi, I have fixed the bug in #2578 , you can have two ways for installation.

1. checkout our repository and install from source with CUDA_EXT=1
2. wait for our pypi release today and pip install from the official PyPI

[fuhengwu2021]

Thank you @FrankLeeeee I actually did the build-from-source step myself before filing this ticket :-) But if other people follow https://github.com/hpcaitech/ColossalAI#install-from-pypi, they will get the same error as above. If they search around, they can find this ticket for reference and work it out :-)

[FrankLeeeee]

That is great, thanks for your efforts!

[FrankLeeeee]

I have added checks before our release to ensure it works for users in #2591 .