This commit is contained in:
parent
e3792ef8e1
commit
460aa98672
|
|
@ -42,7 +42,10 @@ RUN apt-get update && apt-get install -y --no-install-recommends \
|
||||||
|
|
||||||
# ── 安装自编 torch 轮子 ──────────────────────────────────────────────────────
|
# ── 安装自编 torch 轮子 ──────────────────────────────────────────────────────
|
||||||
COPY --from=builder-torch /opt/pytorch/dist /tmp/torch_dist
|
COPY --from=builder-torch /opt/pytorch/dist /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
|
RUN set -e && \
|
||||||
|
echo "==> Files in /tmp/torch_dist:" && ls -lh /tmp/torch_dist && \
|
||||||
|
find /tmp/torch_dist -name 'torch-*.whl' -print | xargs -r python3 -m pip install --no-cache-dir && \
|
||||||
|
rm -rf /tmp/torch_dist
|
||||||
|
|
||||||
|
|
||||||
# ── 编译 torchvision 0.22.1 (依赖本地 torch) ────────────────────────────────
|
# ── 编译 torchvision 0.22.1 (依赖本地 torch) ────────────────────────────────
|
||||||
|
|
|
||||||
Loading…
Reference in New Issue