Skip to content
GitLab
Menu
Projects
Groups
Snippets
Loading...
Help
Help
Support
Community forum
Keyboard shortcuts
?
Submit feedback
Contribute to GitLab
Sign in / Register
Toggle navigation
Menu
Open sidebar
change
sglang
Commits
6f859379
Unverified
Commit
6f859379
authored
Apr 11, 2025
by
Yineng Zhang
Committed by
GitHub
Apr 11, 2025
Browse files
feat: add blackwell workflow (#5303)
parent
f774a0d2
Changes
2
Hide whitespace changes
Inline
Side-by-side
Showing
2 changed files
with
40 additions
and
3 deletions
+40
-3
.github/workflows/release-docker-blackwell.yml
.github/workflows/release-docker-blackwell.yml
+36
-0
docker/Dockerfile.blackwell
docker/Dockerfile.blackwell
+4
-3
No files found.
.github/workflows/release-docker-blackwell.yml
0 → 100644
View file @
6f859379
name
:
Build Blackwell Docker Image
on
:
workflow_dispatch
:
schedule
:
-
cron
:
'
0
0
*
*
*'
jobs
:
build-dev
:
if
:
${{ github.repository == 'sgl-project/sglang' }}
runs-on
:
ubuntu-22.04
steps
:
-
name
:
Checkout repository
uses
:
actions/checkout@v4
-
name
:
Free disk space
uses
:
jlumbroso/free-disk-space@main
with
:
tool-cache
:
false
docker-images
:
false
android
:
true
dotnet
:
true
haskell
:
true
large-packages
:
true
swap-storage
:
false
-
name
:
Login to Docker Hub
uses
:
docker/login-action@v2
with
:
username
:
${{ secrets.DOCKERHUB_USERNAME }}
password
:
${{ secrets.DOCKERHUB_TOKEN }}
-
name
:
Build and Push Blackwell Image
run
:
|
docker build . -f docker/Dockerfile.blackwell -t lmsysorg/sglang:blackwell --no-cache
docker push lmsysorg/sglang:blackwell
docker/Dockerfile.blackwell
View file @
6f859379
...
...
@@ -9,11 +9,12 @@ WORKDIR /sgl-workspace
RUN pip3 install --pre torch torchvision torchaudio --index-url https://download.pytorch.org/whl/nightly/cu128
RUN pip3 install nvidia-nccl-cu12==2.26.2.post1 \
&& pip3 install https://github.com/sgl-project/whl/releases/download/v0.0.8.post1/sgl_kernel-0.0.8.post1+cu128-cp39-abi3-manylinux2014_x86_64.whl \
RUN pip3 install https://github.com/sgl-project/whl/releases/download/v0.0.8.post1/sgl_kernel-0.0.8.post1+cu128-cp39-abi3-manylinux2014_x86_64.whl \
&& pip3 install setuptools==75.0.0 wheel==0.41.0 scikit-build-core
RUN git clone --depth=1 https://github.com/sgl-project/sglang.git \
&& pip3 install -e "python[blackwell]"
&& cd sglang && pip3 install -e "python[blackwell]"
RUN pip3 install nvidia-nccl-cu12==2.26.2.post1 --force-reinstall --no-deps
ENV DEBIAN_FRONTEND=interactive
Write
Preview
Markdown
is supported
0%
Try again
or
attach a new file
.
Attach a file
Cancel
You are about to add
0
people
to the discussion. Proceed with caution.
Finish editing this message first!
Cancel
Please
register
or
sign in
to comment