diff --git a/Dockerfile b/Dockerfile index dae0156b4..e83c220c6 100644 --- a/Dockerfile +++ b/Dockerfile @@ -28,7 +28,8 @@ RUN apt-get update && apt-get install -y --no-install-recommends \ RUN python3 -m pip install --no-cache-dir numpy requests packaging build WORKDIR /opt -RUN git clone --recursive -b v2.7.1 https://github.com/pytorch/pytorch.git +# RUN git clone --recursive -b v2.8.0 https://github.com/pytorch/pytorch.git +COPY ./pytorch_2.8.0/ /opt/pytorch WORKDIR /opt/pytorch ENV MAX_JOBS=${MAX_JOBS} @@ -65,10 +66,10 @@ RUN set -e && \ RUN python3 -c "import torch, typing_extensions, sympy, jinja2, fsspec, networkx; print('✅ Torch:', torch.__version__)" -# ── 编译 torchvision 0.22.1 (依赖本地 torch) ──────────────────────────────── +# ── 编译 torchvision 0.23.0 (依赖本地 torch) ──────────────────────────────── WORKDIR /opt -# RUN git clone -b v0.22.1 https://github.com/pytorch/vision.git -COPY ./vision_0.22.1/ /opt/vision +# RUN git clone -b v0.23.0 https://github.com/pytorch/vision.git +COPY ./vision_0.23.0/ /opt/vision WORKDIR /opt/vision RUN python3 setup.py bdist_wheel && \ pip install --no-cache-dir --no-deps dist/torchvision-*.whl