From e337a6fdd3709cbdf4838533b798377fdf402d3c Mon Sep 17 00:00:00 2001
From: hailin <hailin@gdzx.xyz>
Date: Tue, 1 Jul 2025 12:23:51 +0800
Subject: [PATCH] .

---
 Dockerfile | 2 +-
 1 file changed, 1 insertion(+), 1 deletion(-)

diff --git a/Dockerfile b/Dockerfile
index 212f6bc..4b53120 100644
--- a/Dockerfile
+++ b/Dockerfile
@@ -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