Update lmcache/Dockerfile to CUDA 13.0.1, PyTorch nightly, LMCache dev branch

This commit is contained in:
2026-04-03 08:39:35 +00:00
parent 4980d9e49a
commit 54e609b2c5

View File

@@ -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"]