Skip to content

GitLab

Unverified Commit cadfae66 authored by ishandhanani's avatar ishandhanani Committed by GitHub
Browse files

fix broken deepep/flashmla install in container by adding `--no-build-isolation` (#12170)

parent da1766e4
Hide whitespace changes
Inline Side-by-side
Showing with 2 additions and 2 deletions
docker/Dockerfile
View file @ cadfae66
...@@ -138,7 +138,7 @@ RUN cd /sgl-workspace/DeepEP && \ ...@@ -138,7 +138,7 @@ RUN cd /sgl-workspace/DeepEP && \
echo "Unsupported CUDA version: $CUDA_VERSION" && exit 1 \ echo "Unsupported CUDA version: $CUDA_VERSION" && exit 1 \
;; \ ;; \
esac && \ esac && \
NVSHMEM_DIR=${NVSHMEM_DIR} TORCH_CUDA_ARCH_LIST="${CHOSEN_TORCH_CUDA_ARCH_LIST}" pip install . NVSHMEM_DIR=${NVSHMEM_DIR} TORCH_CUDA_ARCH_LIST="${CHOSEN_TORCH_CUDA_ARCH_LIST}" pip install --no-build-isolation .
# Install flashmla # Install flashmla
RUN git clone https://github.com/deepseek-ai/FlashMLA.git flash-mla && \ RUN git clone https://github.com/deepseek-ai/FlashMLA.git flash-mla && \
...@@ -148,7 +148,7 @@ RUN git clone https://github.com/deepseek-ai/FlashMLA.git flash-mla && \ ...@@ -148,7 +148,7 @@ RUN git clone https://github.com/deepseek-ai/FlashMLA.git flash-mla && \
if [ "$CUDA_VERSION" = "12.6.1" ]; then \ if [ "$CUDA_VERSION" = "12.6.1" ]; then \
export FLASH_MLA_DISABLE_SM100=1; \ export FLASH_MLA_DISABLE_SM100=1; \
fi && \ fi && \
pip install -v . ; pip install --no-build-isolation -v . ;
# Python tools # Python tools
RUN python3 -m pip install --no-cache-dir \ RUN python3 -m pip install --no-cache-dir \
......
Markdown is supported
0% or .
You are about to add 0 people to the discussion. Proceed with caution.
Finish editing this message first!
Please register or to comment