#!/usr/bin/env bash
# aspire2a 环境准备（A100 40G 单卡; python 3.10）
set -euo pipefail
cd "$(dirname "$0")/.."
if ! command -v conda >/dev/null 2>&1; then echo "需要 conda"; exit 1; fi
conda create -n AdcSR python=3.10 -y || true
source "$(conda info --base)/etc/profile.d/conda.sh"
conda activate AdcSR
pip install --upgrade pip
pip install torch torchvision --index-url https://download.pytorch.org/whl/cu121
pip install -r official/requirements.txt
pip install diffusers transformers accelerate peft omegaconf tqdm pyyaml pyiqa lpips scikit-image imageio pillow numpy
pip install huggingface_hub hf_transfer safetensors gdown
pip install "dwt-pytorch" 2>/dev/null || pip install pywavelets || true
python - <<'EOF'
try:
    import pywt; print("pywt ok")
except Exception as e:
    print("pywt missing:", e)
EOF
echo "setup done"