github
/
ColossalAI
mirror of https://github.com/hpcaitech/ColossalAI
1
0
Fork 0
Code Issues Projects Releases Wiki Activity

[docker] fixed ninja build command (#4203)

Browse Source 
* [docker] fixed ninja build command

* polish code
pull/4217/head
Frank Lee 1 year ago committed by GitHub
parent 58913441a1
commit c1cf752021
No known key found for this signature in database
GPG Key ID: 4AEE18F83AFDEB23
1 changed files with 5 additions and 2 deletions
Show Stats Download Patch File Download Diff File

7
docker/Dockerfile
Unescape View File

@ -21,7 +21,10 @@ RUN apt-get update && \
RUN conda install pytorch==1.12.1 torchvision==0.13.1 torchaudio==0.12.1 cudatoolkit=11.3 -c pytorch RUN conda install pytorch==1.12.1 torchvision==0.13.1 torchaudio==0.12.1 cudatoolkit=11.3 -c pytorch
# install ninja # install ninja
RUN apt-get install -y --no-install-recommends ninja-build RUN apt-get update && \
apt-get install -y --no-install-recommends ninja-build && \
apt-get clean && \
rm -rf /var/lib/apt/lists/*
# install apex # install apex
RUN git clone https://github.com/NVIDIA/apex && \ RUN git clone https://github.com/NVIDIA/apex && \
@ -31,7 +34,7 @@ RUN git clone https://github.com/NVIDIA/apex && \
pip install -v --disable-pip-version-check --no-cache-dir --global-option="--cpp_ext" --global-option="--cuda_ext" --global-option="--fast_layer_norm" ./ pip install -v --disable-pip-version-check --no-cache-dir --global-option="--cpp_ext" --global-option="--cuda_ext" --global-option="--fast_layer_norm" ./
# install colossalai # install colossalai
ARG VERSION=1 ARG VERSION=main
RUN git clone -b ${VERSION} https://github.com/hpcaitech/ColossalAI.git \ RUN git clone -b ${VERSION} https://github.com/hpcaitech/ColossalAI.git \
&& cd ./ColossalAI \ && cd ./ColossalAI \
&& CUDA_EXT=1 pip install -v --no-cache-dir . && CUDA_EXT=1 pip install -v --no-cache-dir .

Write Preview
Loading…
Cancel
Save