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