.

2025-07-01 12:23:51 +08:00 · 2025-07-01 12:23:51 +08:00 · e337a6fdd3
parent 1fe51f208d
commit e337a6fdd3
1 changed files with 1 additions and 1 deletions
--- a/2
+++ b/2
@ -15,7 +15,7 @@ RUN apt-get update && apt-get install -y --no-install-recommends \
      libnccl2=2.22.3-1+cuda12.6 \
      libnccl-dev=2.22.3-1+cuda12.6 \
      libjpeg-dev libpng-dev ca-certificates && \
-    python3 -m pip install --no-cache-dir --upgrade pip wheel setuptools sympy pyyaml
+    python3 -m pip install --no-cache-dir --upgrade pip wheel setuptools sympy pyyaml typing-extensions

 WORKDIR /opt
 RUN git clone --recursive -b v2.7.1 https://github.com/pytorch/pytorch.git