From f758f94828ce5439d9d32c5e9937b102f8e66785 Mon Sep 17 00:00:00 2001
From: sora <2075279110@qq.com>
Date: Tue, 28 Jul 2026 02:46:39 +0000
Subject: [PATCH] Update Dockerfile: install torch 2.4.0 (CPU) for tokenizer

---
 tools/docker/Dockerfile.py312 | 6 +++---
 1 file changed, 3 insertions(+), 3 deletions(-)

diff --git a/tools/docker/Dockerfile.py312 b/tools/docker/Dockerfile.py312
index aee6189..434e031 100644
--- a/tools/docker/Dockerfile.py312
+++ b/tools/docker/Dockerfile.py312
@@ -59,9 +59,9 @@ RUN pip install \
     colorlog
 
 # 安装 torch（CPU 版即可，主要用于 tokenizer 加载与部分 benchmark 的 tensor 操作）
-# 固定 numpy==1.26.4 已在前一步完成，torch 2.1.x 与该版本兼容
-RUN pip install torch==2.1.2 --index-url https://download.pytorch.org/whl/cpu 2>/dev/null || \
-    pip install torch==2.1.2 2>/dev/null || true
+# 当前 transformers 要求 torch>=2.4；numpy==1.26.4 仍兼容 torch 2.4.x
+RUN pip install torch==2.4.0 --index-url https://download.pytorch.org/whl/cpu 2>/dev/null || \
+    pip install torch==2.4.0 2>/dev/null || true
 
 # 安装 sandbox 支持（ms-sandbox 等）
 RUN pip install evalscope[sandbox] 2>/dev/null || \