From b5d39f2d1a04c40c926fc4e60e0da91128511788 Mon Sep 17 00:00:00 2001 From: biondizzle Date: Wed, 15 Apr 2026 07:59:17 +0000 Subject: [PATCH] cuda headers --- Dockerfile | 7 ++++++- 1 file changed, 6 insertions(+), 1 deletion(-) diff --git a/Dockerfile b/Dockerfile index a4acb87..eceda3c 100644 --- a/Dockerfile +++ b/Dockerfile @@ -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 \