Skip to content

Commit 9208c9a

Browse files
authored
[Fix] adjust pip install order to keep nccl version as 2.25.1
1 parent 4f3c999 commit 9208c9a

File tree

1 file changed

+2
-2
lines changed

1 file changed

+2
-2
lines changed

Dockerfile

Lines changed: 2 additions & 2 deletions
Original file line numberDiff line numberDiff line change
@@ -38,9 +38,9 @@ WORKDIR /root
3838
COPY ./requirements.txt /lightllm/requirements.txt
3939
RUN pip install -r /lightllm/requirements.txt --no-cache-dir --ignore-installed --extra-index-url https://download.pytorch.org/whl/cu124
4040

41-
RUN pip install --no-cache-dir nvidia-nccl-cu12==2.25.1 # for allreduce hang issues in multinode H100
42-
4341
RUN pip install --no-cache-dir https://github.com/ModelTC/flash-attn-3-build/releases/download/v2.7.4.post1/flash_attn-3.0.0b1-cp39-cp39-linux_x86_64.whl
4442

43+
RUN pip install --no-cache-dir nvidia-nccl-cu12==2.25.1 # for allreduce hang issues in multinode H100
44+
4545
COPY . /lightllm
4646
RUN pip install -e /lightllm --no-cache-dir

0 commit comments

Comments
 (0)