File tree Expand file tree Collapse file tree 1 file changed +0
-6
lines changed
Expand file tree Collapse file tree 1 file changed +0
-6
lines changed Original file line number Diff line number Diff line change @@ -25,9 +25,6 @@ FROM ${BASE_IMAGE} AS install_dependencies
2525
2626WORKDIR /workspace
2727
28- # Might not need to copy cusparseLt in the future once it's included in DLFW cuda container
29- COPY --from=pytorch_image /usr/local/cuda/lib64/libcusparseLt* /usr/local/cuda/lib64/
30-
3128ENV PIP_BREAK_SYSTEM_PACKAGES=1
3229RUN apt-get update -q=2 \
3330 && apt-get install -y --no-install-recommends \
@@ -129,9 +126,6 @@ COPY --from=pytorch_image /usr/local/lib/python3.12/dist-packages/flash_attn /us
129126COPY --from=pytorch_image /usr/local/lib/python3.12/dist-packages/flash_attn-${FLASH_ATTN_VER}.dist-info /usr/local/lib/python3.12/dist-packages/flash_attn-${FLASH_ATTN_VER}.dist-info
130127COPY --from=pytorch_image /usr/local/lib/python3.12/dist-packages/flash_attn_2_cuda.cpython-312-*-linux-gnu.so /usr/local/lib/python3.12/dist-packages/
131128
132- # Might not need to copy cusparseLt in the future once it's included in DLFW cuda container
133- COPY --from=pytorch_image /usr/local/cuda/lib64/libcusparseLt* /usr/local/cuda/lib64/
134-
135129ARG NVRTC_VER
136130ARG NCCL_VER
137131ENV CUDA_VER=$CUDA_VERSION \
You can’t perform that action at this time.
0 commit comments