Skip to content

Commit 5fcf0fa

Browse files
authored
[FIX] specify NVCC_THREADS=1 to pass the docker build in GitHub CI (#816)
1 parent 7068c37 commit 5fcf0fa

File tree

1 file changed

+1
-1
lines changed

1 file changed

+1
-1
lines changed

Dockerfile

Lines changed: 1 addition & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -41,7 +41,7 @@ RUN pip install -r /lightllm/requirements.txt --no-cache-dir --ignore-installed
4141
RUN pip install --no-cache-dir nvidia-nccl-cu12==2.25.1 # for allreduce hang issues in multinode H100
4242

4343
RUN git clone https://github.com/Dao-AILab/flash-attention.git -b v2.7.4.post1
44-
RUN cd flash-attention/hopper && MAX_JOBS=1 NVCC_THREADS=16 python setup.py install
44+
RUN cd flash-attention/hopper && MAX_JOBS=1 NVCC_THREADS=1 python setup.py install
4545

4646
COPY . /lightllm
4747
RUN pip install -e /lightllm --no-cache-dir

0 commit comments

Comments
 (0)