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/common.txt requirements/common.txt
|
||||||
COPY ./vllm_v0.10.0/requirements/cuda.txt requirements/cuda.txt
|
COPY ./vllm_v0.10.0/requirements/cuda.txt requirements/cuda.txt
|
||||||
RUN --mount=type=cache,target=/root/.cache/uv \
|
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 '.')
|
--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'
|
ARG torch_cuda_arch_list='7.0 7.5 8.0 8.9 9.0 10.0 12.0'
|
||||||
|
|
|
||||||
Loading…
Reference in New Issue