[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

@@ -38,7 +38,7 @@ ARG VERSION=main
RUN git clone -b ${VERSION} https://github.com/hpcaitech/ColossalAI.git && \
cd ./ColossalAI && \
git checkout 3e05c07bb8921f2a8f9736b6f6673d4e9f1697d0 && \
CUDA_EXT=1 pip install -v --no-cache-dir .
BUILD_EXT=1 pip install -v --no-cache-dir .
# install titans
RUN pip install --no-cache-dir titans