evalstone/bash/load_swe_images.sh

27 lines
576 B
Bash
Executable File

#!/bin/bash
# 从 save_swe_images.sh 生成的 tar.gz 批量加载 swebench 实例镜像。
set -e
INPUT_DIR="${1:-/data1/sora/evalscope/docker/swe_images}"
if [ ! -d "$INPUT_DIR" ]; then
echo "错误:目录不存在 $INPUT_DIR"
exit 1
fi
TARS=("$INPUT_DIR"/swebench_batch_*.tar.gz)
TOTAL=${#TARS[@]}
echo "共找到 $TOTAL 个镜像包,开始加载..."
IDX=0
for TAR in "${TARS[@]}"; do
IDX=$((IDX + 1))
echo "[$IDX/$TOTAL] 加载 $TAR"
docker load -i "$TAR"
done
echo "完成。已加载镜像数:"
docker images | grep '^swebench/' | wc -l