This commit is contained in:
hailin 2025-08-04 14:36:49 +08:00
parent b21a828e74
commit 1e2a6ad62e
1 changed files with 1 additions and 1 deletions

View File

@ -105,7 +105,7 @@ RUN --mount=type=cache,target=/root/.cache/uv \
COPY ./vllm_v0.10.0/requirements/common.txt requirements/common.txt
COPY ./vllm_v0.10.0/requirements/cuda.txt requirements/cuda.txt
RUN --mount=type=cache,target=/root/.cache/uv \
uv pip install --system -r ./vllm_v0.10.0/requirements/cuda.txt \
uv pip install --system -r requirements/cuda.txt \
--extra-index-url ${PYTORCH_CUDA_INDEX_BASE_URL}/cu$(echo $CUDA_VERSION | cut -d. -f1,2 | tr -d '.')
ARG torch_cuda_arch_list='7.0 7.5 8.0 8.9 9.0 10.0 12.0'