cuda headers

This commit is contained in:
2026-04-15 07:59:17 +00:00
parent 28f9f4c172
commit b5d39f2d1a

View File

@@ -11,7 +11,12 @@ COPY vllm_patches/hf.py /usr/local/lib/python3.12/dist-packages/vllm/renderers/h
# Install LMCache for KV cache offloading / sharing across nodes
# Build with system CUDA 13.0 for Blackwell (B200)
RUN apt-get update && apt-get install -y git && \
RUN apt-get update && apt-get install -y git \
libcusparse-dev-13-0 \
libcublas-dev-13-0 \
libcurand-dev-13-0 \
libcufft-dev-13-0 \
libnvjitlink-dev-13-0 && \
git clone --depth 1 https://github.com/LMCache/LMCache.git /tmp/lmcache && \
cd /tmp/lmcache && \
CUDA_HOME=/usr/local/cuda \