setup_env.sh

#!/bin/bash
# One-click environment setup for C1

set -e

echo "=== Setting up C1 conda environment ==="

# Create or update conda environment
if conda env list | grep -q "^c1 "; then
    echo "Conda environment 'c1' already exists. Activating..."
else
    echo "Creating new conda environment 'c1' with Python 3.12..."
    conda create -n c1 python=3.12 -y
fi

# Activate environment
echo "Activating c1 environment..."
eval "$(conda shell.bash hook)"
conda activate c1

# Install PyTorch with CUDA support
echo "Installing PyTorch with CUDA 12.8..."
pip install torch torchvision torchaudio --index-url https://download.pytorch.org/whl/cu128

# Install verl and llamafactory from git
echo "Installing verl and llamafactory from git..."
pip install --no-deps --no-build-isolation \
    "verl @ git+https://github.com/volcengine/verl.git@facd9fb50193522f87983b89f886afe8c0810acc" \
    "llamafactory @ git+https://github.com/hiyouga/LLaMA-Factory.git@a711bce664faade03b540ad30c41707ba8c928ad"

# Install other dependencies
echo "Installing other dependencies..."
pip install vllm==0.11.0
pip install flash-attn==2.8.1
pip install transformers==4.57.0
pip install "datasets>=2.16.0,<=4.0.0"
pip install python-chess==1.999

echo ""
echo "=== Setup complete! ==="
echo ""
echo "To activate the environment, run:"
echo "  conda activate c1"
echo ""
echo "Don't forget to create api_keys.json with your API keys."