diff --git a/Dockerfile b/Dockerfile index 910fa5b..070ad9b 100644 --- a/Dockerfile +++ b/Dockerfile @@ -42,7 +42,7 @@ 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 $(find /tmp/torch_dist -name "torch-2.7.1+cu126*.whl") && rm -rf /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 # ── 编译 torchvision 0.22.1 (依赖本地 torch) ──────────────────────────────── WORKDIR /opt