upgrade setuptools and wheel if you found "torch module not found" when installing

[MiladInk]

I encountered an issue while installing sglang. After upgrading pip (pip install --upgrade pip), I ran:

pip install "sglang[all]" --find-links https://flashinfer.ai/whl/cu121/torch2.4/flashinfer/

But it failed with the error:
ModuleNotFoundError: No module named 'torch'.

I found on the Flash Attention GitHub that running this solved the issue:

python -m pip install --upgrade pip wheel setuptools

It worked for me, so sharing in case someone faces the same problem! I don't know what the exact reason is though as the error itself was pretty strange.

[zhaochenyang20]

Thanks. We think this may related to the version conflicts between torch, sglang and vllm. @zhyncs