52 lines
1.8 KiB
Docker
52 lines
1.8 KiB
Docker
|
|
# Llama 4 Scout on Iluvatar BI-V150
|
|||
|
|
#
|
|||
|
|
# 基础镜像就是当前 ugly_hopper / test 用的 CoreX vLLM 镜像。
|
|||
|
|
# 这个 Dockerfile 做三件事:
|
|||
|
|
# 1. 把 CoreX vLLM 的 Llama4 chunked-attention 补丁打进镜像
|
|||
|
|
# 2. 安装 entrypoint(容器启动时再次幂等检查补丁)
|
|||
|
|
# 3. 默认以 TP8 / 4096 ctx / 0.90 显存启动 vLLM
|
|||
|
|
#
|
|||
|
|
# 构建:
|
|||
|
|
# docker build -t llama4-scout:2026-10-23 .
|
|||
|
|
#
|
|||
|
|
# 运行(8 卡):
|
|||
|
|
# docker run -d --name llama4-scout \
|
|||
|
|
# --network host --ipc host \
|
|||
|
|
# --device=/dev/iluvatar0 ... --device=/dev/iluvatar7 \
|
|||
|
|
# -v /mnt/disk0/models/LLM-Research/Llama-4-Scout-17B-16E-Instruct:/models/llama4-scout:ro \
|
|||
|
|
# --ulimit memlock=-1 --ulimit stack=67108864 \
|
|||
|
|
# llama4-scout:2026-10-23
|
|||
|
|
#
|
|||
|
|
# 说明:
|
|||
|
|
# - 不需要 --privileged;只需要 GPU 设备节点 + host IPC/network。
|
|||
|
|
# - 如果厂商 runtime 要求整块 /dev,可用 -v /dev:/dev 代替 --device 列表。
|
|||
|
|
# - 模型目录只读挂载,容器内不写模型。
|
|||
|
|
|
|||
|
|
FROM registry.iluvatar.com.cn:10443/customer/sz/vllm0.17.0-4.4.0-x86:v7
|
|||
|
|
|
|||
|
|
LABEL org.opencontainers.image.title="llama4-scout-bi150" \
|
|||
|
|
org.opencontainers.image.description="Llama 4 Scout 17B-16E Instruct on Iluvatar BI-V150 with vLLM 0.17.0+corex" \
|
|||
|
|
org.opencontainers.image.version="2026-10-23"
|
|||
|
|
|
|||
|
|
COPY apply_patch.py /opt/llama4/apply_patch.py
|
|||
|
|
COPY entrypoint.sh /usr/local/bin/llama4-entrypoint.sh
|
|||
|
|
COPY vllm_llama4_chunked_attn.patch /opt/llama4/vllm_llama4_chunked_attn.patch
|
|||
|
|
|
|||
|
|
RUN chmod +x /usr/local/bin/llama4-entrypoint.sh \
|
|||
|
|
&& python3 /opt/llama4/apply_patch.py
|
|||
|
|
|
|||
|
|
ENV MODEL_PATH=/models/llama4-scout \
|
|||
|
|
SERVED_NAME=llama4-scout \
|
|||
|
|
HOST=0.0.0.0 \
|
|||
|
|
PORT=8091 \
|
|||
|
|
TP=8 \
|
|||
|
|
MAX_MODEL_LEN=4096 \
|
|||
|
|
MAX_NUM_SEQS=1 \
|
|||
|
|
GPU_MEM_UTIL=0.90 \
|
|||
|
|
PYTHONUNBUFFERED=1
|
|||
|
|
|
|||
|
|
WORKDIR /opt/llama4
|
|||
|
|
EXPOSE 8091
|
|||
|
|
|
|||
|
|
ENTRYPOINT ["/usr/local/bin/llama4-entrypoint.sh"]
|