Update lmcache/Dockerfile to CUDA 13.0.1, PyTorch nightly, LMCache dev branch
This commit is contained in:
@@ -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"]
|
||||
Reference in New Issue
Block a user