[devops] fix extention building (#5427)

This commit is contained in:
Hongxin Liu
2024-03-05 15:35:54 +08:00
committed by GitHub
parent 822241a99c
commit 070df689e6
24 changed files with 50 additions and 50 deletions

View File

@@ -25,7 +25,7 @@ conda install -c conda-forge cupy cudnn cutensor nccl cuda-version=11.6
# install colossalai with PyTorch extensions
cd <path_to_ColossalAI_repo>
CUDA_EXT=1 pip install -e .
BUILD_EXT=1 pip install -e .
# install other dependencies
pip install triton==2.0.0.dev20221202