This commit is contained in:
parent
b21a828e74
commit
1e2a6ad62e
|
|
@ -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'
|
||||
|
|
|
|||
Loading…
Reference in New Issue