ColossalAI icon indicating copy to clipboard operation
ColossalAI copied to clipboard

[BUG]: Couldn't install cuda_ext

Open ShangWeiKuo opened this issue 2 years ago • 1 comments

🐛 Describe the bug

Excuse me. I installed ColossalAI following the ways you provided with.

After running the command "pip install colossalai==0.1.10+torch1.12cu11.3 -f https://release.colossalai.org", I got the message stated that "If you want to activate cuda mode for MoE, please install with cuda_ext!" and "CUDA_Extension = X".

How should I do to install the correct CUDA_Extension?

MicrosoftTeams-image (1)

Environment

CUDA Version: 11.3 PyTorch Version: 1.12.0+cu113 CUDA Version in PyTorch Build: 11.3 PyTorch CUDA Version Match: Yes CUDA Extension: No

ShangWeiKuo avatar Sep 30 '22 01:09 ShangWeiKuo

Ok, that is a little strange. If the --cuda-ext and --cpp_ext options are working as intended, you should have seen c++ and cuda files being built as part of the install (long gcc and nvcc command lines). Did you see these?

ranjan2829 avatar Oct 01 '22 11:10 ranjan2829

We have updated a lot. This issue was closed due to inactivity. Thanks. https://github.com/hpcaitech/ColossalAI#Installation

binmakeswell avatar Apr 13 '23 04:04 binmakeswell