sora/evalstone
1
1
Fork 0
Code Issues Pull Requests Actions Packages Projects Releases Wiki Activity

Update Dockerfile: install torch 2.4.0 (CPU) for tokenizer

Browse Source
This commit is contained in:
sora 2026-07-28 02:46:39 +00:00
parent ee27130d4d
commit f758f94828
1 changed files with 3 additions and 3 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

6
tools/docker/Dockerfile.py312
View File

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