This commit is contained in:
hailin 2025-07-02 13:18:43 +08:00
parent a0bde3ead4
commit e3792ef8e1
1 changed files with 2 additions and 1 deletions

View File

@ -42,7 +42,8 @@ RUN apt-get update && apt-get install -y --no-install-recommends \
# ── 安装自编 torch 轮子 ──────────────────────────────────────────────────────
COPY --from=builder-torch /opt/pytorch/dist /tmp/torch_dist
RUN python3 -m pip install --no-cache-dir /tmp/torch_dist/torch-2.7.1+cu126*.whl && rm -rf /tmp/torch_dist
RUN sh -c 'python3 -m pip install --no-cache-dir /tmp/torch_dist/torch-2.7.1+cu126*.whl' && rm -rf /tmp/torch_dist
# ── 编译 torchvision 0.22.1 (依赖本地 torch) ────────────────────────────────
WORKDIR /opt