Skip to content

GitLab

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

chore(gb200): remove ToT flashinfer installation (#9079)

parent 89f1d4f5
Hide whitespace changes
Inline Side-by-side
Showing with 0 additions and 8 deletions
docker/Dockerfile.gb200
View file @ 9f24dfef
...@@ -58,8 +58,6 @@ RUN python3 -m pip install --no-cache-dir --upgrade pip setuptools wheel html5li ...@@ -58,8 +58,6 @@ RUN python3 -m pip install --no-cache-dir --upgrade pip setuptools wheel html5li
&& git clone --depth 1 https://github.com/sgl-project/sglang.git \ && git clone --depth 1 https://github.com/sgl-project/sglang.git \
&& cd sglang \ && cd sglang \
&& case "$CUDA_VERSION" in \ && case "$CUDA_VERSION" in \
12.6.1) CUINDEX=126 ;; \
12.8.1) CUINDEX=128 ;; \
12.9.1) CUINDEX=129 ;; \ 12.9.1) CUINDEX=129 ;; \
*) echo "Unsupported CUDA version: $CUDA_VERSION" && exit 1 ;; \ *) echo "Unsupported CUDA version: $CUDA_VERSION" && exit 1 ;; \
esac \ esac \
...@@ -169,12 +167,6 @@ RUN python3 -m pip install --no-cache-dir --break-system-packages \ ...@@ -169,12 +167,6 @@ RUN python3 -m pip install --no-cache-dir --break-system-packages \
matplotlib \ matplotlib \
tabulate tabulate
# Install flashinfer from source to fix a bug
# https://github.com/flashinfer-ai/flashinfer/pull/1413
# FIXME: remove this once flashinfer release > 0.2.10
WORKDIR /sgl-workspace
RUN git clone https://github.com/flashinfer-ai/flashinfer.git --recursive && cd flashinfer && python3 -m pip install -v .
# Install diff-so-fancy # Install diff-so-fancy
RUN curl -LSso /usr/local/bin/diff-so-fancy https://github.com/so-fancy/diff-so-fancy/releases/download/v1.4.4/diff-so-fancy \ RUN curl -LSso /usr/local/bin/diff-so-fancy https://github.com/so-fancy/diff-so-fancy/releases/download/v1.4.4/diff-so-fancy \
&& chmod +x /usr/local/bin/diff-so-fancy && chmod +x /usr/local/bin/diff-so-fancy
......
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