| set -euo pipefail | |
| cd "$(dirname "$0")" | |
| module load Miniconda3 | |
| conda create -y --override-channels -c conda-forge -p "$PWD/.conda" python=3.12 pip | |
| source "$(conda info --base)/etc/profile.d/conda.sh" | |
| conda activate "$PWD/.conda" | |
| python -m pip install torch --index-url https://download.pytorch.org/whl/cu126 | |
| python -m pip install 'transformers==5.22.2' accelerate huggingface_hub safetensors | |
| python -m unittest discover -s tests | |
| python -m pip freeze > environment-resolved.txt | |
| python -c 'import torch; assert torch.cuda.is_available(); print(torch.__version__, torch.cuda.get_device_name(0))' | |