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 || \