Skip to content

fix: Copy cuda libraries from devel to runtime stage #2301

New issue

Have a question about this project? Sign up for a free GitHub account to open an issue and contact its maintainers and the community.

Sign up for GitHub

By clicking “Sign up for GitHub”, you agree to our terms of service and privacy statement. We’ll occasionally send you account related emails.

Already on GitHub? Sign in to your account

Merged
nv-tusharma merged 7 commits into from
Aug 8, 2025
Merged

fix: Copy cuda libraries from devel to runtime stage #2301

Changes from all commits
Commits
Show all changes
7 commits
Select commit Hold shift + click to select a range
37a3b49
Copy CUDA directory into runtime container
nv-tusharma Aug 5, 2025
1b38124
Merge branch 'main' into tusharma/add-cuda-libs-sglang
nv-tusharma Aug 6, 2025
15c1003
Merge branch 'main' into tusharma/add-cuda-libs-sglang
nv-tusharma Aug 6, 2025
a9731fa
Only copy required cuda libraries for flashinfer installation
nv-tusharma Aug 7, 2025
bd03999
Merge branch 'main' into tusharma/add-cuda-libs-sglang
nv-tusharma Aug 7, 2025
61ea92f
Merge branch 'main' into tusharma/add-cuda-libs-sglang
nv-tusharma Aug 7, 2025
0af0c68
Merge branch 'main' into tusharma/add-cuda-libs-sglang
nv-tusharma Aug 7, 2025
File filter

Filter by extension

Filter by extension
Conversations
Failed to load comments.
Loading
Jump to
Jump to file
Failed to load files.
Loading
Diff view
Diff view
12 changes: 11 additions & 1 deletion container/Dockerfile.sglang
View file
Open in desktop
Original file line number Diff line number Diff line change
Expand Up @@ -404,7 +404,8 @@ ENV PATH="${VIRTUAL_ENV}/bin:${PATH}"
# Copy nats and etcd from base image
COPY --from=base /usr/bin/nats-server /usr/bin/nats-server
COPY --from=base /usr/local/bin/etcd/ /usr/local/bin/etcd/
ENV PATH=/usr/local/bin/etcd/:$PATH
# Add ETCD and CUDA binaries to PATH so cicc and other CUDA tools are accessible
ENV PATH=/usr/local/bin/etcd/:/usr/local/cuda/nvvm/bin:$PATH

# Copy UCX from base image as plugin for NIXL
# Copy NIXL source from wheel_builder image
Expand All @@ -416,6 +417,15 @@ ENV NIXL_PLUGIN_DIR=$NIXL_LIB_DIR/plugins
COPY --from=base /usr/local/ucx /usr/local/ucx
COPY --from=wheel_builder $NIXL_PREFIX $NIXL_PREFIX

# Copy CUDA development tools (nvcc, headers, dependencies, etc.) from base devel image
COPY --from=base /usr/local/cuda/bin/nvcc /usr/local/cuda/bin/nvcc
COPY --from=base /usr/local/cuda/bin/cudafe++ /usr/local/cuda/bin/cudafe++
COPY --from=base /usr/local/cuda/bin/ptxas /usr/local/cuda/bin/ptxas
COPY --from=base /usr/local/cuda/bin/fatbinary /usr/local/cuda/bin/fatbinary
COPY --from=base /usr/local/cuda/include/ /usr/local/cuda/include/
COPY --from=base /usr/local/cuda/nvvm /usr/local/cuda/nvvm
COPY --from=base /usr/local/cuda/lib64/libcudart.so* /usr/local/cuda/lib64/

ENV LD_LIBRARY_PATH=\
$NIXL_LIB_DIR:\
$NIXL_PLUGIN_DIR:\
Expand Down
Loading