From 54e609b2c56725c960da2148649877a24270147e Mon Sep 17 00:00:00 2001 From: biondizzle Date: Fri, 3 Apr 2026 08:39:35 +0000 Subject: [PATCH] Update lmcache/Dockerfile to CUDA 13.0.1, PyTorch nightly, LMCache dev branch --- lmcache/Dockerfile | 9 ++++++++- 1 file changed, 8 insertions(+), 1 deletion(-) diff --git a/lmcache/Dockerfile b/lmcache/Dockerfile index 5f2860b..24ba7d9 100644 --- a/lmcache/Dockerfile +++ b/lmcache/Dockerfile @@ -39,7 +39,8 @@ ENV LD_LIBRARY_PATH=${CUDA_HOME}/lib64:${LD_LIBRARY_PATH} # Install Python deps in venv RUN uv pip install numpy==2.0.0 -RUN uv pip install torch==2.8.0 torchvision torchaudio --extra-index-url https://download.pytorch.org/whl/cu129 +# Install PyTorch nightly with CUDA 13.0 (bleeding edge) +RUN uv pip install --pre torch torchvision torchaudio --index-url https://download.pytorch.org/whl/nightly/cu130 FROM base AS build-base @@ -52,4 +53,10 @@ RUN git clone https://github.com/LMCache/LMCache.git -b v0.3.3 && \ uv pip install setuptools_scm && \ python -m build --wheel --no-isolation && \ cp dist/*.whl /workspace/ +CMD ["/bin/bash"]">>> COMMIT: $(git rev-parse HEAD)" && \ + echo ">>> DATE: $(git log -1 --format=%cd --date=short)" && \ + echo "========================================\n\n" && \ + uv pip install setuptools_scm && \ + python -m build --wheel --no-isolation && \ + cp dist/*.whl /workspace/ CMD ["/bin/bash"] \ No newline at end of file