File size: 913 Bytes
3dac39e | 1 2 3 4 5 6 7 8 9 10 11 12 13 14 15 16 17 18 19 20 21 22 23 24 25 26 27 28 29 30 | #!/usr/bin/env bash
# Setup script for a fresh Vast.ai instance.
# Assumes: PyTorch + CUDA image, Ubuntu, this repo uploaded to ~/alkyline
set -euo pipefail
echo "=== Arcspan Vast.ai Setup ==="
cd ~/alkyline
# 1. Create venv and install OPF
python3 -m venv .venv
source .venv/bin/activate
pip install --upgrade pip
pip install -e vendor/privacy-filter/
# 2. Verify GPU
python3 -c "import torch; print(f'CUDA: {torch.cuda.is_available()}, Device: {torch.cuda.get_device_name(0) if torch.cuda.is_available() else \"none\"}')"
# 3. Download base checkpoint (auto on first run, but let's trigger it)
echo "Downloading base checkpoint..."
opf "test" > /dev/null 2>&1 || true
echo ""
echo "=== Setup complete. Run training with: ==="
echo " source .venv/bin/activate"
echo " bash scripts/train_cyner.sh 40 cuda 8"
echo ""
echo "Then evaluate with:"
echo " bash scripts/eval_cyner.sh checkpoints/cyner_v1 cuda"
|