mirror of
https://git.datalinker.icu/vllm-project/vllm.git
synced 2026-09-08 21:07:06 +08:00
Merge ff00ef309dc619752c4c22028583e36a6f7e37a1 into 254f6b986720c92ddf97fbb1a6a6465da8e87e29
This commit is contained in:
commit
333b646717
@ -51,6 +51,8 @@ FROM fetch_vllm_${REMOTE_VLLM} AS fetch_vllm
|
|||||||
# -----------------------
|
# -----------------------
|
||||||
# vLLM build stages
|
# vLLM build stages
|
||||||
FROM fetch_vllm AS build_vllm
|
FROM fetch_vllm AS build_vllm
|
||||||
|
# Set target device to ROCm for build
|
||||||
|
ENV VLLM_TARGET_DEVICE=rocm
|
||||||
# Build vLLM
|
# Build vLLM
|
||||||
RUN cd vllm \
|
RUN cd vllm \
|
||||||
&& python3 -m pip install -r requirements/rocm.txt \
|
&& python3 -m pip install -r requirements/rocm.txt \
|
||||||
|
|||||||
Loading…
x
Reference in New Issue
Block a user