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

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

diff --git a/Dockerfile b/Dockerfile
index 1cd4961..212f6bc 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
+    python3 -m pip install --no-cache-dir --upgrade pip wheel setuptools sympy pyyaml
 
 WORKDIR /opt
 RUN git clone --recursive -b v2.7.1 https://github.com/pytorch/pytorch.git