#!/bin/bash
# ============================================================
# 把本地新镜像 scp 到远程机器并加载
# 用法: bash scripts/copy_image_to_machine.sh <IP>
# ============================================================

set -euo pipefail

IP="${1:?请提供远程机器 IP}"
PASS="${SSHPASS:-sskj2025}"
LOCAL_TAR="/data1/sora/evalscope/docker/evalscope-complete-py312.tar.gz"
REMOTE_TAR="/data1/sora/evalscope/docker/evalscope-complete-py312.tar.gz"

echo "==> [$IP] 复制镜像 tar 包..."
sshpass -p "$PASS" scp -o StrictHostKeyChecking=no "$LOCAL_TAR" "root@$IP:$REMOTE_TAR"

echo "==> [$IP] 加载 Docker 镜像..."
sshpass -p "$PASS" ssh -o StrictHostKeyChecking=no "root@$IP" "docker load -i $REMOTE_TAR"

echo "==> [$IP] 验证 torch..."
sshpass -p "$PASS" ssh -o StrictHostKeyChecking=no "root@$IP" "docker run --rm evalscope-complete-py312:latest python -c 'import torch; print(\"torch\", torch.__version__)'"

echo "==> [$IP] 完成"