Skip to content

Commit 3187a9d

Browse files
committed
add libnuma dependency
1 parent d554c58 commit 3187a9d

File tree

1 file changed

+13
-10
lines changed

1 file changed

+13
-10
lines changed

Dockerfile

Lines changed: 13 additions & 10 deletions
Original file line numberDiff line numberDiff line change
@@ -103,37 +103,40 @@ USER modelrunner
103103

104104
# Install uv and vLLM as modelrunner user
105105
RUN curl -LsSf https://astral.sh/uv/install.sh | sh \
106-
&& ~/.local/bin/uv venv --python /usr/bin/python3 /opt/vllm-env \
107-
&& if [ "$TARGETARCH" = "amd64" ]; then \
108-
WHEEL_ARCH="manylinux_2_31_x86_64"; \
109-
WHEEL_URL="https://github.com/vllm-project/vllm/releases/download/v${VLLM_VERSION}/vllm-${VLLM_VERSION}%2B${VLLM_CUDA_VERSION}-${VLLM_PYTHON_TAG}-${WHEEL_ARCH}.whl"; \
110-
~/.local/bin/uv pip install --python /opt/vllm-env/bin/python "$WHEEL_URL"; \
106+
&& ~/.local/bin/uv venv --python /usr/bin/python3 /opt/vllm-env \
107+
&& if [ "$TARGETARCH" = "amd64" ]; then \
108+
WHEEL_ARCH="manylinux_2_31_x86_64"; \
109+
WHEEL_URL="https://github.com/vllm-project/vllm/releases/download/v${VLLM_VERSION}/vllm-${VLLM_VERSION}%2B${VLLM_CUDA_VERSION}-${VLLM_PYTHON_TAG}-${WHEEL_ARCH}.whl"; \
110+
~/.local/bin/uv pip install --python /opt/vllm-env/bin/python "$WHEEL_URL"; \
111111
else \
112-
~/.local/bin/uv pip install --python /opt/vllm-env/bin/python "vllm==${VLLM_VERSION}"; \
112+
~/.local/bin/uv pip install --python /opt/vllm-env/bin/python "vllm==${VLLM_VERSION}"; \
113113
fi
114114

115115
RUN /opt/vllm-env/bin/python -c "import vllm; print(vllm.__version__)" > /opt/vllm-env/version
116116

117117
# --- SGLang variant ---
118118
FROM llamacpp AS sglang
119119

120-
ARG SGLANG_VERSION=0.4.0
120+
ARG SGLANG_VERSION=0.5.6
121121

122122
USER root
123123

124-
RUN apt update && apt install -y python3 python3-venv python3-dev curl ca-certificates build-essential && rm -rf /var/lib/apt/lists/*
124+
RUN apt update && apt install -y python3 python3-venv python3-dev curl ca-certificates build-essential libnuma1 libnuma-dev numactl && rm -rf /var/lib/apt/lists/*
125125

126126
RUN mkdir -p /opt/sglang-env && chown -R modelrunner:modelrunner /opt/sglang-env
127127

128128
USER modelrunner
129129

130130
# Install uv and SGLang as modelrunner user
131131
RUN curl -LsSf https://astral.sh/uv/install.sh | sh \
132-
&& ~/.local/bin/uv venv --python /usr/bin/python3 /opt/sglang-env \
133-
&& ~/.local/bin/uv pip install --python /opt/sglang-env/bin/python "sglang==${SGLANG_VERSION}"
132+
&& ~/.local/bin/uv venv --python /usr/bin/python3 /opt/sglang-env \
133+
&& ~/.local/bin/uv pip install --python /opt/sglang-env/bin/python "sglang==${SGLANG_VERSION}"
134134

135135
RUN /opt/sglang-env/bin/python -c "import sglang; print(sglang.__version__)" > /opt/sglang-env/version
136136

137+
# Verify aiohttp is installed (required by sglang server)
138+
RUN /opt/sglang-env/bin/python -c "import aiohttp"
139+
137140
FROM llamacpp AS final-llamacpp
138141
# Copy the built binary from builder
139142
COPY --from=builder /app/model-runner /app/model-runner

0 commit comments

Comments
 (0)