Add dev-deepep docker image (#6198)
This commit is contained in:
36
.github/workflows/release-docker-dev-deepep.yml
vendored
Normal file
36
.github/workflows/release-docker-dev-deepep.yml
vendored
Normal file
@@ -0,0 +1,36 @@
|
||||
name: Build Dev-DeepEP 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 DeepEP Image
|
||||
run: |
|
||||
docker build . -f docker/Dockerfile.dev-deepep -t lmsysorg/sglang:dev-deepep --no-cache
|
||||
docker push lmsysorg/sglang:dev-deepep
|
||||
80
docker/Dockerfile.dev-deepep
Normal file
80
docker/Dockerfile.dev-deepep
Normal file
@@ -0,0 +1,80 @@
|
||||
FROM lmsysorg/sglang:dev
|
||||
|
||||
# CMake
|
||||
RUN apt-get update \
|
||||
&& apt-get install -y --no-install-recommends \
|
||||
build-essential \
|
||||
wget \
|
||||
libssl-dev \
|
||||
&& wget https://github.com/Kitware/CMake/releases/download/v3.27.4/cmake-3.27.4-linux-x86_64.sh \
|
||||
&& chmod +x cmake-3.27.4-linux-x86_64.sh \
|
||||
&& ./cmake-3.27.4-linux-x86_64.sh --skip-license --prefix=/usr/local \
|
||||
&& rm cmake-3.27.4-linux-x86_64.sh
|
||||
|
||||
# Python
|
||||
RUN apt-get update \
|
||||
&& apt-get install -y --no-install-recommends \
|
||||
python3 \
|
||||
python3-pip \
|
||||
&& ln -s /usr/bin/python3 /usr/bin/python
|
||||
|
||||
# GDRCopy
|
||||
WORKDIR /tmp
|
||||
RUN git clone https://github.com/NVIDIA/gdrcopy.git
|
||||
WORKDIR /tmp/gdrcopy
|
||||
RUN git checkout v2.4.4
|
||||
|
||||
RUN apt update
|
||||
RUN apt install -y nvidia-dkms-535
|
||||
RUN apt install -y build-essential devscripts debhelper fakeroot pkg-config dkms
|
||||
RUN apt install -y check libsubunit0 libsubunit-dev
|
||||
|
||||
WORKDIR /tmp/gdrcopy/packages
|
||||
RUN CUDA=/usr/local/cuda ./build-deb-packages.sh
|
||||
RUN dpkg -i gdrdrv-dkms_*.deb
|
||||
RUN dpkg -i libgdrapi_*.deb
|
||||
RUN dpkg -i gdrcopy-tests_*.deb
|
||||
RUN dpkg -i gdrcopy_*.deb
|
||||
|
||||
ENV GDRCOPY_HOME=/usr/src/gdrdrv-2.4.4/
|
||||
|
||||
# IBGDA dependency
|
||||
RUN ln -s /usr/lib/x86_64-linux-gnu/libmlx5.so.1 /usr/lib/x86_64-linux-gnu/libmlx5.so
|
||||
RUN apt-get install -y libfabric-dev
|
||||
|
||||
# DeepEP
|
||||
WORKDIR /sgl-workspace
|
||||
RUN git clone https://github.com/deepseek-ai/DeepEP.git
|
||||
|
||||
# NVSHMEM
|
||||
WORKDIR /sgl-workspace
|
||||
RUN wget https://developer.download.nvidia.com/compute/redist/nvshmem/3.2.5/source/nvshmem_src_3.2.5-1.txz
|
||||
RUN tar -xf nvshmem_src_3.2.5-1.txz \
|
||||
&& mv nvshmem_src nvshmem
|
||||
|
||||
WORKDIR /sgl-workspace/nvshmem
|
||||
RUN git apply /sgl-workspace/DeepEP/third-party/nvshmem.patch
|
||||
|
||||
WORKDIR /sgl-workspace/nvshmem
|
||||
ENV CUDA_HOME=/usr/local/cuda
|
||||
RUN NVSHMEM_SHMEM_SUPPORT=0 \
|
||||
NVSHMEM_UCX_SUPPORT=0 \
|
||||
NVSHMEM_USE_NCCL=0 \
|
||||
NVSHMEM_MPI_SUPPORT=0 \
|
||||
NVSHMEM_IBGDA_SUPPORT=1 \
|
||||
NVSHMEM_PMIX_SUPPORT=0 \
|
||||
NVSHMEM_TIMEOUT_DEVICE_POLLING=0 \
|
||||
NVSHMEM_USE_GDRCOPY=1 \
|
||||
cmake -S . -B build/ -DCMAKE_INSTALL_PREFIX=/sgl-workspace/nvshmem/install -DCMAKE_CUDA_ARCHITECTURES=90 \
|
||||
&& cd build \
|
||||
&& make install -j
|
||||
|
||||
WORKDIR /sgl-workspace/DeepEP
|
||||
ENV NVSHMEM_DIR=/sgl-workspace/nvshmem/install
|
||||
RUN NVSHMEM_DIR=/sgl-workspace/nvshmem/install pip install .
|
||||
|
||||
RUN set -euxo ; \
|
||||
curl --proto '=https' --tlsv1.2 -sSf https://just.systems/install.sh | bash -s -- --to /usr/local/bin
|
||||
|
||||
# Set workspace
|
||||
WORKDIR /sgl-workspace
|
||||
Reference in New Issue
Block a user