BUILD_EXT=1 pip install colossalai